Algorithm/dynamic programming

[백준] 알고리즘 1003번 - 피보나치 함수 문제

낭강 2021. 3. 9. 22:52
문제

다음 소스는 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이 각각 몇 번 출력되는지 구하는 프로그램을 작성하시오.

 

소스코드
#include <bits/stdc++.h>
using namespace std;
int dp[41];
int main(){
   ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
   int T;
   cin >> T;
   dp[0] = 0, dp[1] = 1, dp[2] = 1;
   for (int i = 3; i <= 40; i++) dp[i] = dp[i - 2] + dp[i - 1];
   while (T--) {
	   int n;
	   cin >> n;
	   if (n == 0) cout << "1 0";
	   else if (n == 1) cout << "0 1";
	   else cout << dp[n - 1] << " " << dp[n];
	   cout << "\n";
   }
}
풀이

피보나치 4까지만 해봐도 대충 규칙이 있다는 것을 알 수 있습니다.

우선 구해야하는건 fb(1)과 fb(0)이 나타나는 개수를 출력하는 건데요.

      0      1       2       3     4

fb   0       1      1       2     3

0    1       0      1       1     2

1    0       1      1       2     3

 

fb(0)은 자기 라인의 피보나치를 이루고 있고

fb(1)도 마찬가지로 자기 라인의 피보나치를 이루고 있다.

 

즉 n에 따라 0은 n의 피보나치 수 - 1 

1은 n의 피보나치와 같으므로

 

출력을 dp[n-1] dp[n] 을 해주면 됩니다.