[백준/파이썬] 7770번 아즈텍 피라미드

-코드

n = int(input())
h, block = 1, 0
while block <= n:
    block += h ** 2 + (h - 1) ** 2
    h += 1
print(h - 2)

어떤 문제인지 이해를 못 해서 질문 게시판을 참고하여 풀었습니다.