[백준 브론즈 1] 피보나치 수 2 - kotlin, DP

[Bronze I] 피보나치 수 2 - 2748

문제 링크

성능 요약

메모리: 12316 KB, 시간: 88 ms

분류

다이나믹 프로그래밍(dp), 수학(math)

문제 설명

피보나치 수는 0과 1로 시작한다. 0번째 피보나치 수는 0이고, 1번째 피보나치 수는 1이다. 그다음 2번째부터는 바로 앞 두 피보나치 수의 합이 된다.

이를 식으로 써보면 Fn = Fn-1 + Fn-2 (n ≥ 2)가 된다.

n=17일때 까지 피보나치 수를 써보면 다음과 같다.

0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233, 377, 610, 987, 1597

n이 주어졌을 때, n번째 피보나치 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 n이 주어진다. n은 90보다 작거나 같은 자연수이다.

출력

첫째 줄에 n번째 피보나치 수를 출력한다.

풀이

val list = LongArray(91) { 0}

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()


    bw.write("${fibo(br.readLine().toInt())}")
    bw.flush()
}

fun fibo(n: Int): Long {
    if (n <= 2) return 1
    if (list[n] != 0.toLong()) return list[n]
    list[n] = fibo(n - 1) + fibo(n - 2)
    return list[n]

}

이 문제는 dp로 풀어야 하는 문제이다.
먼저 n의 크기가 90보다 작거나 같다고 했으므로 그 크기만큼 배열을 만들어준다.
피보나치 함수에는 만약 n이 2보다 작으면 1이므로 return 1을 해주고
만약 배열의 n index값이 채워져 있다면 이미 함수가 실행되었으므로 그 배열의 인덱스 값을 return 해준다.
배열의 n 값에 재귀함수를 돌아 n-1 + n-2 값을 넣어준다.