[백준] 9461번 파도반 수열 (파이썬)

9461번: 파도반 수열

 

코드

import sys

n = int(input())
li = list(sys.stdin.readline().rsplit() for i in range(n))

def count(l):
    base = [1,1,1,2,2,3]
    answer = 0
    for i in range(5,l-1):
        answer = base[i] + base[i-4]
        base.append(answer)
    print(base[l-1])

for i in li:
    count(int(i[0]))