Algorithm/Baekjoon

9461번 파이썬

@Eeap 2022. 4. 9. 09:42
반응형

이 문제는 초기 설정만 하면 규칙이 일정한 문제이다.

규칙은 i 번째 값이 i-1번쨰와 i-5번째를 더한 값이 반복되는 구조이다.

즉 1~5까지는 초기 설정해주고 나머지는 규칙을 이용한 dp를 이용해서 풀면 쉽게 풀리는 문제인것 같다!

import sys
input=sys.stdin.readline
t = int(input())
p=[0 for _ in range(101)]
p[1]=1
p[2]=1
p[3]=1
p[4]=2
p[5]=2
for i in range(6,101):
    p[i] = p[i-1]+p[i-5]
for _ in range(t):
    n = int(input())
    print(p[n])
반응형