백준 알고리즘 4922번 Walk Like an Egyptian(python)

단순 수학 문제이다.

while 1:
    n = int(input())
    if n == 0:
        break
    print(f"{n} => {n**2 - (n-1)}")