[Kotlin] 백준 10870 - 피보나치 수 5

문제

피보나치 수는 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은 20보다 작거나 같은 자연수 또는 0이다.

풀이

1. n번째 자리의 값은 n-1번째, n-2번째 값의 합이다. -> n-1, n-2번째 값을 기억하고 있어야한다. -> 배열 자료구조를 채택해서 값을 저장해 놓는다.

2. n < 20의 조건 -> 0부터 n까지 모두 보아도 충분히 작은 값, 시간복잡도 O(n) 이므로 하나씩 보는 방식을 채택한다.

3. n이 0과 1의 경우는 예외처리한다.

코드

import java.util.*

fun main() = with(Scanner(System.`in`)) {
    val n: Int = nextInt()
    val N: ArrayList<Int> = ArrayList<Int>()
    
    for(i in 0..n){
        if(i == 0){
            N.add(0)
        } else if(i == 1){
           N.add(1) 
        } else {
            N.add(N.get(i-1) + N.get(i-2))
        }
    }
    
    println(N.get(n))
}