diff --git a/Week03/pyramid_sema_erol.py b/Week03/pyramid_sema_erol.py index ac816c41..0817100b 100644 --- a/Week03/pyramid_sema_erol.py +++ b/Week03/pyramid_sema_erol.py @@ -1,10 +1,11 @@ def calculate_pyramid_height(number_of_blocks): height = 0 - required_blocks = 1 - - while number_of_blocks >= required_blocks: - number_of_blocks -= required_blocks + blocks_used = 0 + level = 1 + + while blocks_used + level <= number_of_blocks: + blocks_used += level height += 1 - required_blocks += 1 - + level += 1 + return height