알고리즘/DP

[Python] 백준 10844 쉬운 계단 수

JayAlex07 2023. 3. 24. 14:40

🧑‍💻 [Python] 백준 10844 쉬운 계단 수

Silver 1 - DP

img

 

 

코드

N = int(input())

stairs = [[0] * 10 for _ in range(N)]

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

for i in range(1, N):
    for j in range(10):
        if j == 0:
            stairs[i][j] = stairs[i-1][j + 1]

        elif 0 < j < 9:
            stairs[i][j] = stairs[i-1][j+1] + stairs[i-1][j-1]

        elif j == 9:
            stairs[i][j] = stairs[i-1][j-1]


print(sum(stairs[N-1]) % 1000000000)