[Python] 백준 9461번 - 파도반 수열

문제

https://www.acmicpc.net/problem/9461

 

---

파도반 수열은 다음과 같은 특성을 갖고 있음.

=> p(i) = p(i-2) + p(i-3)

 

표준 입력을 위해 sys 모듈 추가.

테스트 케이스 수 입력.

최대 100개 항까지 표현하므로 항 번호와 인덱스 번호를 일치시키기 위해 101개의 0을 갖는 리스트 새엇ㅇ.

1번째, 2번째, 3번째 항의 값을 1로 초기화.

import sys


input = sys.stdin.readline
t = int(input())
arr = [0] * 101
arr[1] = arr[2] = arr[3] = 1

 

4번째부터 100번째까지 항 계산.

for i in range(4, 101):
    arr[i] = arr[i-2] + arr[i-3]

 

테스트 케이스 수 만큼 반복.

- 정수 하나 입력.

- 해당 정수 번째 항 출력,

for _ in range(t):
    n = int(input())
    print(arr[n])