[프로그래머스] 피보나치 수 / Kotlin

https://programmers.co.kr/learn/courses/30/lessons/12945

코딩테스트 연습 - 피보나치 수

 

문제

피보나치 수는 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을 완성해 주세요.

제한 사항

• n은 2 이상 100,000 이하인 자연수입니다.

입출력 예

n	return
3	2
5	5

입출력 예 설명

피보나치수는 0번째부터 0, 1, 1, 2, 3, 5, ... 와 같이 이어집니다.

---

 

의식의 흐름

1. 재귀로 풀 수도 있으며 시간복잡도는 N(2^100,000)까지 갈 수 있다.

2. 따라서 초기화한 배열에 저장해두고(메모이제이션) 꺼내서 리턴해주는것이 낫다.

3. intArray이므로 처음부터 %로 나눠서 저장해야 한다.

 

해결 코드

class Solution {
    fun solution(n: Int): Int {
        
        var arr = IntArray(n + 1){1}
        arr[0] = 0
        
        if (n > 2) {
            (3..n).forEach { arr[it] = (arr[it-1] + arr[it-2]) % 1234567 }
        }
        
        return arr[n]
    }
}