[BOJ] 1,2,3 더하기(python)

문제 링크 - https://www.acmicpc.net/problem/9095

사고 과정

• 문제를 보고 product를 사용하면 될 것 같았다.
• product로 만들 수 있는 경우의 수를 다 만들고 각각의 합이 n과 같으면 cnt에 +1을 해주었다.

나의 풀이

import sys
from itertools import product
t = int(sys.stdin.readline())
a = [1,2,3]

for _ in range(t):
    n = int(sys.stdin.readline())
    cnt = 0
    for i in range(1, n+1):
        a_list = list(product(a, repeat = i))
        for j in range(len(a_list)):
            if sum(a_list[j]) == n:
                cnt +=1
    print(cnt)