본문 바로가기
알고리즘/[python] 백준 BOJ

[Python] 백준 15998 1,2,3 더하기 3

by Alan_Kim 2022. 12. 25.
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

댓글