백준 알고리즘 7770번 아즈텍 피라미드(python)

재밌는 수학 문제이다. (h*2-1)**2 - 4*(h*(h-1)//2)가 핵심인 것 같다. 더 좋은 수식이 있을 것 같기도 하고... 

n = int(input())
x = h = 0
while x <= n:
    h += 1
    x += (h*2-1)**2 - 4*(h*(h-1)//2)
if x > n:
    h -= 1
print(h)