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 |
반응형