From d0bbde4a39e805630028d4be99860031152ef508 Mon Sep 17 00:00:00 2001
From: Samuel Vermeulen <samuel@vermeulen.pro>
Date: Thu, 5 Jan 2023 00:04:49 +0100
Subject: [PATCH] maj

---
 pyramide.py | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/pyramide.py b/pyramide.py
index f289b63..4d30651 100644
--- a/pyramide.py
+++ b/pyramide.py
@@ -1,13 +1,10 @@
 nbPierres = int(input())
 pierre = 0
-surEtage = 0
-nbTotalPierres = 0
+etage = 1
 
-while nbPierres > 0:
-	pierre +=1
-	surfEtage = pierre ** 2
-	nbTotalPierres = nbTotalPierres + surfEtage
-	nbPierres = nbPierres - nbTotalPierres
+while pierre + etage ** 2 <= nbPierres:
+	pierre = pierre + etage ** 2
+	etage +=1
 
-print(pierre)
-print(nbTotalPierres)
+print(etage - 1)
+print(pierre)
\ No newline at end of file