본문 바로가기
백준알고리즘/동적 계획법1

(Python/🥇5)백준알고리즘 15989번: 1, 2, 3 더하기 4

by windy7271 2024. 8. 2.
728x90
반응형

 

문제 바로가기 

문제:

정수 4를 1, 2, 3의 합으로 나타내는 방법은 총 4가지가 있다. 합을 나타낼 때는 수를 1개 이상 사용해야 한다. 합을 이루고 있는 수의 순서만 다른 것은 같은 것으로 친다.

 

  • 1+1+1+1
  • 2+1+1 (1+1+2, 1+2+1)
  • 2+2
  • 1+3 (3+1)

정수 n이 주어졌을 때, n을 1, 2, 3의 합으로 나타내는 방법의 수를 구하는 프로그램을 작성하시오.

입력:

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다. n은 양수이며 10,000보다 작거나 같다.

출력:

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

 

풀이:

import sys

sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

t = int(input())
dp = [1] * 10001

for i in range(2, 10001):
    dp[i] += dp[i - 2]
for i in range(3, 10001):
    dp[i] += dp[i - 3]

for _ in range(t):
    print(dp[int(input())])

 

 

순서가 다르면 그것도 같은 것이다. 라는게 좀 어려웠다.

 

이것도 노트에 끄적여 보면

 

각 숫자마다 1 로 이루어진 정답이 있으니 dp 모든 구간을 1로 만들어준다.

1 = 1

2 = 1+1

3 = 1+1+1

4 = 1+1+1+1

5 = 1+1+1+1+1

 

이런식으로, 그리고

2개전 dp[i-2] 에서 + 2 해서 만들 수 있고

3개전 dp[i-3] 에서 + 3 으로 만들 수 있으니

이것을 for 문으로 만들어 준다.

 

 

반응형

댓글