Bonjour, je suis en première NSI
j'ai cette exercice a faire mais j'ai du mal avancer.

Une pyramide à base carrée, de hauteur h, est constituée avec des cubes identiques de la manière
suivante :
- h=1: 1 étage, 1 cube
- h=2 : 2 étages, 5 cubes (4 +1)
- h=3 : 3 étages, 14 cubes (5+9)
Écrire un programme qui demande le nombre de cubes dont on dispose et qui renvoie la hauteur de la pyramide ainsi que le nombre de cubes qu’il reste.

J'ai penser trouver une équation permettant trouver les étages suivant, j'ai trouver
La relation de la pyramide est par exemple h3+h3-h2+h3=h4 soit 14+14-5+14=37
h4=37 mais pour le reste donc taper le programme et mettre en place l'équation je n'y arrive pas. Merci d'avance je commence a péter les plombs sur cette exo.