본문 바로가기

코딩 테스트/Dynamic Programming

[프로그래머스] 피보나치 수(Lv 2) with Kotlin

문제

피보나치 수는 F(0) = 0, F(1) = 1일 때, 1 이상의 n에 대하여 F(n) = F(n-1) + F(n-2) 가 적용되는 수

  • F(2) = F(0) + F(1) = 0 + 1 = 1
  • F(3) = F(1) + F(2) = 1 + 1 = 2
  • F(4) = F(2) + F(3) = 1 + 2 = 3
  • F(5) = F(3) + F(4) = 2 + 3 = 5

2 이상의 n이 입력되었을 때, n번째 피보나치 수를 1234567으로 나눈 나머지를 리턴하는 함수, solution을 완성해 주세요.

 

풀이 과정

피보나치 수는 다이나믹 프로그래밍의 대표적인 문제이므로 DP를 사용하여 해결한다.

 

코드

class Solution {
    fun solution(n: Int): Int {
        val dp = Array<Int>(n+1){0}
        
        dp[0] = 0
        dp[1] = 1
        
        for(i in 2..n){
            dp[i] = (dp[i-1]+dp[i-2])%1234567
        }
        
        return dp[n]
    }
}