728x90
반응형
https://www.acmicpc.net/problem/15988
15988번: 1, 2, 3 더하기 3
각 테스트 케이스마다, n을 1, 2, 3의 합으로 나타내는 방법의 수를 1,000,000,009로 나눈 나머지를 출력한다.
www.acmicpc.net
문제해결
DP를 공부했으면 굉장히 쉽게 풀었을 수 있다.
n을 숫자 1,2,3의 합으로 나타내는 방법의 수를 dp[n]이라고 하면,
dp[n] = dp[n-1] + dp[n-2] + dp[n-3]
왜냐하면 n-1를 숫자 1,2,3의 합으로 나타낸 후 숫자 1을 더한 것의 개수는 dp[n-1]과 동일
n-2를 숫자 1,2,3의 합으로 나타낸 후 숫자 2를 더한 것의 개수는 dp[n-2]과 동일하며
n-3을 숫자 1,2,3의 합으로 나타낸 후 숫자 3을 더한 것의 개수는 dp[n-3]과 동일하기 때문이다.
CODE
import sys
input = sys.stdin.readline
t = int(input())
dp = [0 for _ in range(1000001)]
dp[1] = 1
dp[2] = 2
dp[3] = 4
dp[4] = 7 # dp[3] + dp[2] + dp[1]
for i in range(5, 1000001):
dp[i] = dp[(i-1)]%1000000009 + dp[(i-2)]%1000000009 + dp[(i-3)]%1000000009
# 이렇게 안하면 메모리 초과가 뜬다.ㅜㅜ
for _ in range(t):
n = int(input())
print(dp[n]%1000000009)
728x90
반응형
'알고리즘 > [python] 백준 BOJ' 카테고리의 다른 글
[Python] 백준 11057 오르막 수 (0) | 2022.12.26 |
---|---|
[Python] 백준 1309 동물원 (0) | 2022.12.25 |
[Python] 백준 1699 제곱수의 합 (0) | 2022.12.25 |
[Python] 백준 2225 합분해 (0) | 2022.12.24 |
[Python] 백준 2193 이친수 (0) | 2022.12.22 |
댓글