다음 소스는 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이 각각 몇 번 출력되는지 구하는 프로그램을 작성하시오.
첫째 줄에 테스트 케이스의 개수 T가 주어진다.
각 테스트 케이스는 한 줄로 이루어져 있고, N이 주어진다. N은 40보다 작거나 같은 자연수 또는 0이다.
3
0
1
3
1 0
0 1
1 2
2
6
22
5 8
10946 17711
피보나치 수열을 먼저 알아야함. 물론 코드가 주어졌기 때문에 코드를 보면 알 수 있지만 배경지식을 알면 도움이 된다.
피보나치 수열은 첫째 및 둘째 항이 1이고 그 뒤의 모든 항은 바로 앞 두 항의 합인 수열이다. 보통 0번째 항은 편의상 0으로 세팅한다.
0 1 1 2 3 ...
-> 2는 전의 숫자(1) + 그 전의 숫자(1)
-> 3은 전의 숫자(2) + 그 전의 숫자(1)
.
.
.
이렇게 이뤄 진다.
문제에서는 0과 1의 개수를 묻기 때문에 세팅을 다르게 해줘야 한다.
t = int(input())
for _ in range(t):
zero, one = 1, 0
n = int(input())
for i in range(n):
zero, one = one, zero+one
print(zero, one)
위에서 zero = 0, one = 1로 초기화를 해줬다. 다시 말하지만 값을 원하는 것이 아니라 개수를 구해야 하기 때문에 초기 상태는 이렇게 해놓는다.
zero, one = one, zero+one
이 부분은 0의 개수는 1이 나오는 개수와 동일하고 반대로
1의 개수는 0이 나온 개수와 1이 나온 개수를 더한 만큼 나온다.