[Python/BOJ] 2688. 줄어들지 않아

https://www.acmicpc.net/problem/2688

문제

어떤 숫자가 줄어들지 않는다는 것은 그 숫자의 각 자리 수보다 그 왼쪽 자리 수가 작거나 같을 때 이다.

예를 들어, 1234는 줄어들지 않는다. 

줄어들지 않는 4자리 수를 예를 들어 보면 0011, 1111, 1112, 1122, 2223이 있다. 줄어들지 않는 4자리수는 총 715개가 있다.

이 문제에서는 숫자의 앞에 0(leading zero)이 있어도 된다. 0000, 0001, 0002는 올바른 줄어들지 않는 4자리수이다.

n이 주어졌을 때, 줄어들지 않는 n자리 수의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T(1 <= T <= 1,000)이 주어진다. 각 테스트 케이스는 숫자 하나 n으로 이루어져 있다. (1 <= n <= 64)

출력

각 테스트 케이스에 대해 한 줄에 하나씩 줄어들지 않는 n자리 수의 개수를 출력한다.

풀이

Code

import sys
input = sys.stdin.readline

# 1. 2차원 dp 생성
dp = [ [0 for _ in range(10)] for _ in range(65)]
# 2. 초기값 설정
dp[1] = [1 for _ in range(10)]
# 3.
for i in range(2, 65) :
    for j in range(10) :
        # 3-1. 점화식에 따라 처리
        dp[i][j] = sum(dp[i-1][:j+1])
# 4. 결과 리스트 생성
result = [sum(row) for row in dp]
# 5.
for _ in range(int(input())) :
    # 5-1. 결과 출력
    print(result[int(input())])