11057 - 오르막 수

8642 단어 baekjoonDPplzrunDP

📚 11057 - 오르막 수

오르막 수
기존 나의 코드

n = int(input())
INF = 10007

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

dp[n] = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]


for i in range(n-1, -1, -1):
    for j in range(0, 10):
        cur_data = 0
        for z in range(j, 10):
            cur_data += dp[i+1][z]
            cur_data %= INF
        dp[i][j] = cur_data


print(dp[0][0])


 

상위권 효율적인 코드

- arr[0][0~9]는 다 1로
- arr[k][9] : 9로 시작한다면 다 1이다.
- arr[i][j] = arr[i][j+1] + arr[i-1][j]
- 최종 arr[n+1][0] : n+1번째 0번 인덱스를 출력하면 된다.

 

n = int(input())

arr = [[0] * 10 for _ in range(n + 2)]

# n + 1번째의 0번째 값이 이전 n 번째까지 구한 총합이 된다.
for i in range(0, n + 2):
    arr[i][9] = 1

for i in range(1, n + 2):
    for j in range(8, -1, -1):
        arr[i][j] = arr[i][j + 1] + arr[i - 1][j]

print(arr[n + 1][0] % 10007)

 

결과

  • 규칙을 잡아 DP를 이용하니 for문 2개로 해결되었다.
  • 시간이 대략 12ms 차이난다.

 

좋은 웹페이지 즐겨찾기