[Swift] 알고리즘 공부 - 피보나치 수

문제 설명

피보나치 수는 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

 

설명

1. n이 0일경우와 1일경우를 제외하고 나머지 수는 재귀함수로 구현

func solution(_ n:Int) -> Int {
    return fibonacci(n) % 1234567
}

func fibonacci(_ n: Int) -> Int {
    if n == 0 { return 0 }
    else if n == 1 { return 1 }
    return fibonacci(n - 1) + fibonacci(n-2)
}

 

피보나치수 문제는 보통 재귀함수로 문제를 해결한 경우가 있었는데 이 문제의 경우 재귀함수로 구현하면 시간초과가 뜬다. 그래서 반복문으로 다시 구현했다.

func solution(_ n:Int) -> Int {
    var fibonacci: [Int] = [0, 1]

    if n == 0 || n == 1 {
        return n
    }

    for i in 2...n {
        fibonacci.append((fibonacci[i-2] + fibonacci[i-1]) % 1234567)
    }

    return fibonacci[n]
}