CS/알고리즘_문제풀이(파이썬)

1, 2, 3 더하기

Jedy_Kim 2021. 7. 12. 18:39
728x90

문제

정수 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의 합으로 나타내는 방법의 수를 구하는 프로그램을 작성하시오.

입력

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

출력

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

예제 입력 1 

3

4

7

10

예제 출력 1 복사

7

44

274

 

#코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import sys
input=sys.stdin.readline
 
def dpMakeNum(n):
    if(n>3):dptable=[0]*(1+n)
    else:dptable=[0]*(1+3)
    dptable[1]=1
    dptable[2]=2
    dptable[3]=4
    # an = a_n-1 + a_n-2 + a_n-3
    for i in range(4,n+1):
        dptable[i]=(dptable[i-1]+dptable[i-2]+dptable[i-3])%1000007
    return dptable[n]
 
    
if __name__=="__main__":
    n = int(input())
    for _ in range(n):
      get_n = int(input())
      print(dpMakeNum(get_n))
cs

 

반응형

'CS > 알고리즘_문제풀이(파이썬)' 카테고리의 다른 글

직사각형의 합  (0) 2021.07.13
그룹 단어 체커  (0) 2021.07.12
2×n 타일링  (0) 2021.07.12
타겟 넘버  (0) 2021.07.10
트리에서의 거리  (0) 2021.07.07