문제
다음 소스는 N번째 피보나치 수를 구하는 C++ 함수이다.
1 2 3 4 5 6 7 8 9 10 11 |
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)을 호출하면 다음과 같은 일이 일어난다.
- fibonacci(3)은 fibonacci(2)와 fibonacci(1) (첫 번째 호출)을 호출한다.
- fibonacci(2)는 fibonacci(1) (두 번째 호출)과 fibonacci(0)을 호출한다.
- 두 번째 호출한 fibonacci(1)은 1을 출력하고 1을 리턴한다.
- fibonacci(0)은 0을 출력하고, 0을 리턴한다.
- fibonacci(2)는 fibonacci(1)과 fibonacci(0)의 결과를 얻고, 1을 리턴한다.
- 첫 번째 호출한 fibonacci(1)은 1을 출력하고, 1을 리턴한다.
- fibonacci(3)은 fibonacci(2)와 fibonacci(1)의 결과를 얻고, 2를 리턴한다.
1은 2번 출력되고, 0은 1번 출력된다. N이 주어졌을 때, fibonacci(N)을 호출했을 때, 0과 1이 각각 몇 번 출력되는지 구하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다.
각 테스트 케이스는 한 줄로 이루어져 있고, N이 주어진다. N은 40보다 작거나 같은 자연수 또는 0이다.
출력
각 테스트 케이스마다 0이 출력되는 횟수와 1이 출력되는 횟수를 공백으로 구분해서 출력한다.
나의 답
#include <bits/stdc++.h>
using namespace std;
// N이 0~40까지 이므로
int table_N[41] = {0, 1, };
// 계산시간을 단축하기 위해 기존에 계산했던 값은 바로 리턴하도록 함
// 0이나 1일 때는 -1을 참조하기 때문에 그 경우는 제외함
int fibonacci(int n) {
if(table_N[n]==0 && n>1)
table_N[n] = fibonacci(n-1) + fibonacci(n-2);
return table_N[n];
}
int main(void) {
int T, N;
cin >> T;
for(int i = 0; i < T; ++i) {
cin >> N;
if(N>=1) {
if(N>1)
fibonacci(N);
cout << table_N[N-1] << ' ' << table_N[N] << endl;
}
else
cout << 1 << ' ' << 0 << endl;
}
return 0;
}
'백준' 카테고리의 다른 글
[10845번] 큐 (0) | 2020.05.09 |
---|---|
[2869번] 달팽이는 올라가고 싶다 (0) | 2020.05.04 |
[1463번] 1로 만들기 (0) | 2020.05.03 |
[16431번] 베시와 데이지 (0) | 2020.05.03 |
[9461번] 파도반 수열 (0) | 2020.05.03 |