[백준][1003번: 피보나치 함수]

문제

문제링크

다음 소스는 N번째 피보나치 수를 구하는 C++ 함수이다.

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이 출력되는 횟수를 공백으로 구분해서 출력한다.

접근방법

처음에는 fibonacci 함수를 만들어 재귀를 돌면서 0을 return 할 때마다 0의 개수 증가, 1을 return 할 때마다 1의 개수 증가로 풀었다. 하지만 시간초과가 났다....
그래서 dp(Dynamic Programming) 방법으로 zero(0)의 배열과 one(1)의 배열을 만들어서 구했다.

코드(시간초과)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    static int T;
    static int one;
    static int zero;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        T = Integer.parseInt(br.readLine());

        for(int i=0; i<T; i++){
            int n = Integer.parseInt(br.readLine());
            fibonacci(n);
            System.out.println(zero + " " + one);
            one = 0;
            zero = 0;
        }
    }
    static int fibonacci(int n){
        if(n==0){
            zero++;
            return 0;
        }
        if(n==1){
            one++;
            return 1;
        }
        return fibonacci(n-1)+fibonacci(n-2);
    }
}

코드(성공)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    static int T;
    static int[] zero;
    static int[] one;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        T = Integer.parseInt(br.readLine());
        zero = new int[41];
        one = new int[41];
        zero[0] = 1; //0일 때 0의 개수
        one[0] = 0; // 0일 때 1의 개수
        zero[1] = 0; // 1일 때 0의 개수
        one[1] = 1; // 1일 때 1의 개수
        StringBuilder sb = new StringBuilder();
        for(int i=0; i<T; i++){
            int n = Integer.parseInt(br.readLine());
            for(int j=2; j<=n; j++){
                zero[j] = zero[j-1]+ zero[j-2];
                one[j] = one[j-1]+ one[j-2];
            }
            sb.append(zero[n]).append(" ").append(one[n]).append('\n');
        }
        System.out.println(sb.toString());
    }
}