[Python/Kotlin] 백준 24416번 : 알고리즘 수업 - 피보나치 수 1

백준 문제 주소
https://www.acmicpc.net/problem/24416

문제

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

오늘은 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)이 주어진다.

5

30

출력

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

5 3

832040 28

---

Python으로 푸는 것이 더 편하지만 코테를 위해서 Kotlin으로도 푸는 연습을 해야하기 때문에 두 가지 언어로 문제를 풀 예정

Python 풀이

def fib(n):
    if n == 1 or n == 2:
        return 1
    else:
        return fib(n-1) + fib(n-2)

def fibonacci(n):
    f = [0]*(n+1)
    f[1], f[2] = 1, 1
    count = 0
    for i in range(3, n+1):
        f[i] = f[i-1] + f[i-2]
        count += 1
    return count

n = int(input())

print(fib(n), fibonacci(n))

문제의 의사코드를 참고하여 코드를 구현했습니다.

재귀를 사용하기 때문에 시간초과가 발생한 것 같아서 pypy3으로 제출했더니 통과했습니다

Kotlin 풀이

fun main() {
	val value = readLine()!!.toInt()
    val fib = fib(value)
    val fibonacci = fibonacci(value)
    print("$fib $fibonacci")
}

fun fib(n: Int): Int {
    if(n==1 || n==2) {
        return 1
    }
    else {
        return (fib(n-1) + fib(n-2))
    }
}

fun fibonacci(n: Int): Int {
    var count = 0
    var array = Array(n+1, {0})
    array[1] = 1
    array[2] = 1
    
    for(i in 3..n) {
        array[i] = array[i-1] + array[i-2]
        count += 1
    }
    return count
}