알고리즘 도전기 - 90

10844 쉬운 계단 수

문제

45656이란 수를 보자.
이 수는 인접한 모든 자리의 차이가 1이다. 이런 수를 계단 수라고 한다.
N이 주어질 때, 길이가 N인 계단 수가 총 몇 개 있는지 구해보자. 0으로 시작하는 수는 계단수가 아니다.

입력

첫째 줄에 N이 주어진다. N은 1보다 크거나 같고, 100보다 작거나 같은 자연수이다.

출력

첫째 줄에 정답을 1,000,000,000으로 나눈 나머지를 출력한다.

예제 입력

2

예제 출력

17

코드

import sys

N = int(sys.stdin.readline().rstrip())
memo = [0, 1, 1, 1, 1, 1, 1, 1, 1, 1]

for i in range(2, N + 1):
    fake_arr = [0] * 10
    for j in range(len(fake_arr)):
        if j == 0:
            fake_arr[1] += memo[j]
        elif j == 9:
            fake_arr[8] += memo[9]
        else:
            fake_arr[j - 1] += memo[j]
            fake_arr[j + 1] += memo[j]
    for j in range(10):
        memo[j] = fake_arr[j]
result = 0
for e in memo:
    result += e
print(result % 1000000000)