[백준] 9050 - 1,2,3 더하기 (py)

youznn·2023년 2월 13일
0

문제

정수 4를 1, 2, 3의 합으로 나타내는 방법은 총 7가지가 있다. 합을 나타낼 때는 수를 1개 이상 사용해야 한다.

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

코드

import sys

t = int(input())

dic ={1:[[1]],2:[[2],[1,1]],3:[[1,1,1],[1,2],[2,1],[3]]}

for i in range (4,12):
    for j in dic[i-1]:
        dic[i] = dic.get(i,[])+[j + [1]]
    for j in dic[i-2]:
        if [j+[1]] not in dic[i]:
            dic[i] = dic.get(i,[])+[j + [1]]
    for j in dic[i-3]:
        if [j + [1]] not in dic[i]:
            dic[i] = dic.get(i,[])+[j + [1]]

for n in range (t):
    n = int(sys.stdin.readline())
    print(len(dic[n]))

풀이

DP에 대해 어느정도 감이 익어가는 중이다...
일단 문제 조건을 보면 n은 11 이하의 양수라고 되어 있으므로, 총 경우의 수를 다 따져서 list나 dict에 저장해놓고 쓰면 편하다.

다른 사람들 코드를 보니... 내가 상당히 복잡하게 풀었음을 알 수 있었다. 사실 그냥 개수만 구하면 되는 거라서 일일히 세부 사항들을 고려할 필요가 없다.

profile
https://github.com/youznn

0개의 댓글