QA & Engineering Blog

A Blog about Quality · Automation · Engineering

🏠 홈으로

[Silver I] 쉬운 계단 수 - 10844

문제 링크

성능 요약

메모리: 113112 KB, 시간: 116 ms

분류

다이나믹 프로그래밍

문제 설명

45656이란 수를 보자.

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

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

입력

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

출력

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

💡 Solutions

📄 쉬운 계단 수.py

# import sys
N = int(input())

dp = [[0 for _ in range(10)] for _ in range(N+1)]

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

for i in range(2,N+1):
  for j in range(10):
    if j == 0:
      dp[i][j] = dp[i-1][1]
    elif j == 9:
      dp[i][j] = dp[i-1][8]
    else:
      dp[i][j] = (dp[i-1][j-1] + dp[i-1][j+1])

print(sum(dp[N])%1000000000)
## 이전 결과가 이후의 값에 영향을 미침
# dp = [0 for _ in range(N+1)]
# dp[1] = 9 
# dp[2] = 17 # (9-2)*2 + 2 * 2 - 1
# dp[3] = 3  # (17-2)*2 + 2 * 2

# for i in range(3,N+1):
#   dp[i] = (dp[i-1]*2 + 2*2 - (i+1)%2)%1000000000
# print(dp)