[프로그래머스] 피보나치 수 by Python

문제

피보나치 수는 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 이하인 자연수입니다.

입출력 예

풀이

피보나치 수는 다이나믹 프로그래밍(dp)의 전형적인 기초 문제이다.
재귀로도 간단하게 풀 수 있지만 효율성으로 생각했을 때는 배열에 저장하면서 이전의 값들을 활용하는 것이 훨씬 효율적이기 때문에 for문과 배열을 이용해서 문제를 풀이했다.

def solution(n):
    answer = 0
    arr = [] # dp로 활용되며 수를 저장할 배열
    arr.append(0)
    arr.append(1)
    for i in range(2, n+1): # 2부터 시작하여 전과 전전의 수를 활용
        arr.append(arr[i-1] + arr[i-2])
    answer = arr[n] % 1234567
    return answer