[BOJ/백준] 4922 - Walk Like an Egyptian [python]

문제 링크

4922번: Walk Like an Egyptian

문제 풀이

수열의 규칙을 찾기 쉬운 문제였다. n번째 수는 모두 대각선에 위치해 있다.

[1, 3, 7, 13] 순으로 수열이 진행됨을 확인할 수 있다.

이 수열은 n이 커질수록 차가 2씩 증가하고 있다.

• 1과 3의 차 2
• 3과 7의 차 4
• 7과 13의 차 6
• ...

이를 코드로 작성해 풀 수 있었다.

정답 코드

lst = [0] * 1001
lst[1] = 1
lst[2] = 3
for i in range(3, 1000) : lst[i] = lst[i - 1] + lst[i - 1] - lst[i - 2] + 2
while True :
    n = int(input())
    if n == 0 : break
    print("{:d} => {:d}".format(n, lst[n]))