[Baekjoon] 10844번 쉬운 계단 수

문제

45656이란 수를 보자.

이 수는 인접한 모든 자리수의 차이가 1이 난다. 이런 수를 계단 수라고 한다.

세준이는 수의 길이가 N인 계단 수가 몇 개 있는지 궁금해졌다.

N이 주어질 때, 길이가 N인 계단 수가 총 몇 개 있는지 구하는 프로그램을 작성하시오. (0으로 시작하는 수는 없다.)

입력

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

출력

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

예제

Example 1:

Input: 
1
Output: 
9

Example 2:

Input:
2
Output:
17

조건

시간 제한 : 1초

메모리 제한 : 256 MB

풀이과정

내 풀이

1자리일 때 0에서 9까지 개수를 리스트에 담는다. 그리고 for문을 이용해서 값을 구하게 되는데 새로운 리스트 arr2를 생성하고 여기에 0과 9의 경우에는 인접한 수가 1과 8밖에 없으므로 1과 8에 해당하는 개수를 업데이트해주고 나머지의 경우에는 양쪽으로 인접한 수의 합을 업데이트해준다.

그 후 arr을 arr2로 바꿔주면서 이 과정을 i = n이 나올때까지 반복한다.

# n = 계단 수
n = int(input())

# 1자리일 때 0~9 개수
arr = [0,1,1,1,1,1,1,1,1,1]

# dp 테이블 정의
dp = [0]*101
dp[1] = 9

# 값 계산 및 출력
for i in range(2, n+1):
    arr2 = [0]*10
    for j in range(10):
        if j == 0:
            arr2[j] = arr[j+1]
        elif j == 9:
            arr2[j] = arr[j-1]
        else:
            arr2[j] = arr[j-1]+arr[j+1]
    arr = arr2
print(sum(arr)%1000000000)