[ Python ] 오르막수

 

python을 잘 사용하지 못해서 시작을 어떻게 할지에 대해 열심히 고민했다.

 

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

해당 코드를 통해 2차원 리스트를 생성할 수 있다. 문제를 풀 때 출력에서 가장 큰 고생을 해서 여러 방면으로 출력을 해보려고 노력했다.

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

python에서는 배열의 index에 마이너스 기호(-)를 붙이면 뒤에서부터 index를 가르킨다.

단순히 dp[-1] 이라고 선언하면 dp 배열의 마지막행 전체를 나타낸다.

 

 

Code

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

for r in range(1, N):
    for c in range(1, 10):
        dp[r][c] = dp[r][c-1] + dp[r-1][c]

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