다음 소스는 N번째 피보나치 수를 구하는 C++ 함수이다.
int fibonacci(int n) {
if (n == 0) {
printf("0");
return 0;
} else if (n == 1) {
printf("1");
return 1;
} else {
return fibonacci(n‐1) + fibonacci(n‐2);
}
}
fibonacci(3)을 호출하면 다음과 같은 일이 일어난다.
1은 2번 출력되고, 0은 1번 출력된다.
N이 주어졌을 때, fibonacci(N)을 호출했을 때, 0과 1이 각각 몇 번 출력되는지 구하는 프로그램을 작성하시오.
시간초과
import sys
input = sys.stdin.readline
t = int(input())
def fibonacci(n):
global print_0, print_1
if n == 0:
print_0 += 1
return 0
elif n == 1:
print_1 += 1
return 1
else:
return fibonacci(n-1) + fibonacci(n-2)
for _ in range(t):
print_0 = 0
print_1 = 0
n = int(input())
fibonacci(n)
print(print_0, print_1)
정답
zero = [1,0,1]
one = [0,1,1]
def fibonacci(n):
length = len(zero)
if n >= length:
for i in range(length, n+1):
zero.append(zero[i-1] + zero[i-2])
one.append(one[i-1] + one[i-2])
print('{} {}'.format(zero[n],one[n]))
T = int(input())
for _ in range(T):
fibonacci(int(input()))