[백준/10844] 쉬운 계단 수 (Python)

Problem : https://www.acmicpc.net/problem/10844

10844번: 쉬운 계단 수

 

Difficulty : Silver 1

 

Status : Solved

 

Time : 00:04:13

 

---

 

문제 설명

 

45656이란 수를 보자.

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

N이 주어질 때, 길이가 N인 계단 수가 총 몇 개 있는지 구해보자. 0으로 시작하는 수는 계단수가 아니다.

 

입력 및 출력

 

입력

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

 

출력

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

 

입력 예시

1

 

출력 예시

 

9

 

 

---

 

풀이

 

DP 문제. 초기값 설정을 할 때, 0을 제외한 모든 값을 1로 설정하는 것만 주의하면 된다. i의 길이를 가진 계단 수의 마지막 숫자를 j라고 할때, 그 경우의 수는 i-1 길이를 가지고 마지막 숫자가 j-1 혹은 j+1인 직전의 계단수의 경우의 수의 합이 된다.

 

 

풀이 코드

N = int(input())
MOD = 1000000000
dp = [[0]*10 for _ in range(N)]

for i in range(1, 10) :
  dp[0][i] = 1

for i in range(1, N) :
  for j in range(10) :
    if j-1 >= 0 :
      dp[i][j] = (dp[i][j] + dp[i-1][j-1]) % MOD
    if j+1 < 10 :
      dp[i][j] = (dp[i][j] + dp[i-1][j+1]) % MOD

print(sum(dp[-1]) % MOD)

풀이 완료~