본문 바로가기
공부/알고리즘

[알고리즘][백준] 15988번 1, 2, 3 더하기 3 - 파이썬(python) 풀이

by 무명오리 2024. 3. 11.

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

 

15988번: 1, 2, 3 더하기 3

각 테스트 케이스마다, n을 1, 2, 3의 합으로 나타내는 방법의 수를 1,000,000,009로 나눈 나머지를 출력한다.

www.acmicpc.net


9095번 1, 2, 3 더하기의 응용문제인데 n의 값이 11보다 작다에서 1,000,000보다 작거나 같다로 변경된 것으로 확인!

+) 15990번 1, 2, 3 더하기 5도 존재

import sys

t = int(sys.stdin.readline())
num = []

for _ in range(t):
    num.append(int(sys.stdin.readline()))

dp = [0] * 10000001
dp[1] = 1
dp[2]  = 2
dp[3] = 4

for i in range(4, max(num)+1):
    dp[i] = (dp[i-1] + dp[i-2] + dp[i-3]) % 1000000009

for i in num:
    print(dp[i])

 

굳이 숫자들을 리스트에 넣어 max(num)을 구하지 않고 

for i in range(4, 1000001):

 

위와 같은 형태로 하는 것이 메모리 사용량도 덜하니 참고