백준: 16195 1, 2, 3 더하기 9

1, 2, 3 더하기 시리즈

• 1, 2, 3 더하기
• 1, 2, 3 더하기 2
• 1, 2, 3 더하기 3
• 1, 2, 3 더하기 4
• 1, 2, 3 더하기 5
• 1, 2, 3 더하기 6
• 1, 2, 3 더하기 7
• 1, 2, 3 더하기 8
• 1, 2, 3 더하기 9

 

문제

16195번: 1, 2, 3 더하기 9

 

문제 풀이

 앞서 다룬 1, 2, 3 더하기 7을 이해하고 있다면 쉽게 풀 수 있는 문제이다. 1, 2, 3 더하기 7에서는 N을 구할 때 M 만큼 사용 가능한 경우만 구하면 된다. 이 문제에서는 M이하의 모든 경우를 구하여야 하기에 1, 2, 3 더하기 7에서 M이하의 모든 경우를 더해주면 문제에서 원하는 답안을 구할 수 있다.

 

코드

from sys import stdin


if __name__ == '__main__':
    MAX = 1001
    divisor = 1000000009
    dp = [[0] * MAX for _ in range(MAX)]

    # 1을 나타낼 수 있는 경우
    dp[1][1] = 1
    # 2를 나타낼 수 잇는 겨우
    dp[2][1] = 1
    dp[2][2] = 1
    # 3을 나타낼 수 있는 경우
    dp[3][1] = 1
    dp[3][2] = 2
    dp[3][3] = 1

    for i in range(4, MAX):
        # 사용할 수 있는 수는 현재수의 크기 만큼이다.
        # 예를 들어 5의 경우 숫자 6개로 사용하여 만들 수 있는 경우는 없다.
        for j in range(1, i + 1):
            dp[i][j] = (dp[i - 1][j - 1] + dp[i - 2][j - 1] + dp[i - 3][j - 1]) % divisor

    t = int(stdin.readline())

    for _ in range(t):
        n, m = map(int, stdin.readline().split())
        print(sum(dp[n][1:m + 1]) % divisor)