백준1003 피보나치 함수(파이썬)

08 Aug 2022 in Study on Algorithm

백준1003 피보나치 함수(파이썬)

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

문제

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다.
각 테스트 케이스는 한 줄로 이루어져 있고, N이 주어진다. N은 40보다 작거나 같은 자연수 또는 0이다.

출력

각 테스트 케이스마다 0이 출력되는 횟수와 1이 출력되는 횟수를 공백으로 구분해서 출력한다.

코드

import sys
sys.setrecursionlimit(10**9)
t=int(input())
while t:
    n=int(input())
    zero=[1,0,1]
    one=[0,1,1]
    if n>=3:
        for i in range(3,n+1):
            zero.append(zero[i-2]+zero[i-1])
            one.append(one[i-2]+one[i-1])
    print(zero[n],one[n])
    t-=1