[백준/파이썬]9507번 Generations of Tribbles

-코드

def sol(n):
    if n < 2:
        return 1
    elif n == 2:
        return 2
    elif n == 3:
        return 4
    elif n > 3:
        return arr[n - 1] + arr[n - 2] + arr[n - 3] + arr[n - 4]


for _ in range(int(input())):
    arr = []
    for i in range(int(input()) + 1):
        arr.append(sol(i))
    print(arr[-1])