[백준] B1 24416번 알고리즘 수업 - 피보나치 수 1 (JAVA)

문제 출처 - Baekjoon Online Judge

문제는 여기

24416번: 알고리즘 수업 - 피보나치 수 1

[문제] 

오늘도 서준이는 동적 프로그래밍 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해보자.

오늘은 n의 피보나치 수를 재귀호출과 동적 프로그래밍으로 구하는 알고리즘을 배웠다. 재귀호출에 비해 동적 프로그래밍이 얼마나 빠른지 확인해 보자. 아래 의사 코드를 이용하여 n의 피보나치 수를 구할 경우 코드1 코드2 실행 횟수를 출력하자.

피보나치 수 재귀호출 의사 코드는 다음과 같다.

fib(n) {
    if (n = 1 or n = 2)
    then return 1;  # 코드1
    else return (fib(n - 1) + fib(n - 2));
}

피보나치 수 동적 프로그래밍 의사 코드는 다음과 같다.

fibonacci(n) {
    f[1] <- f[2] <- 1;
    for i <- 3 to n
        f[i] <- f[i - 1] + f[i - 2];  # 코드2
    return f[n];
}

[입력]

첫째 줄에 n(5 ≤ n ≤ 40)이 주어진다.

[출력]

코드1 코드2 실행 횟수를 한 줄에 출력한다.

 

 

---

[풀이]

1. 주어진 fib 함수를 만들어준다.

2. 주어진 fibonacci 함수를 만들어준다.

3. fib의 경우 n이 1, 2가 되면 해당 재귀 수행을 종료하는 것이기 때문에 카운트를 증가시켜 준다.

4. dp를 이용하여 수행 횟수를 dp 배열에 담아준다.

5. 해당 결과들을 출력해 준다.

[접근]

1. fib 함수와 fibonacci 함수를 만들어 각각 몇 번의 실행을 하는지 구해서 출력하면 되겠다고 생각하였다.

[코드]

import java.io.*;
import java.util.*;

public class Main {
	static int cnt1;
	static int cnt2;
	static int dp[];

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int n = Integer.parseInt(br.readLine());
		dp = new int[n + 1];
		
		fib(n);
		fibonacci(n);
		
		System.out.println(cnt1 + " " + cnt2);
	}

	public static int fib(int n) {
		// 종료조건
		if (n == 1 || n == 2) {
			cnt1++; // 재귀를 멈추기 전에 cnt 증가
			return 1;
		}
		else
			return (fib(n - 1) + fib(n - 2)); // 다음 재귀 실행
	}
	
	public static int fibonacci(int n) {
		// 초기화
		dp[0] = 1;
		dp[1] = 1;

                // dp 배열 채우기
                for (int i = 2; i < n; i++) {
                    cnt2++; // 횟수 증가
                    dp[i] = dp[i - 2] + dp[i - 1];
                }

                return dp[n - 1];
	}
}