[Algorithm] Programmers : 피보나치 수 by Python

[문제 바로가기] 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은 1이상, 100000이하인 자연수입니다.

입출력 예
n return
3 2
5 5
입출력 예 설명
피보나치수는 0번째부터 0, 1, 1, 2, 3, 5, ... 와 같이 이어집니다.

💡 문제 풀이

피보나치 수를 구하는 코딩만 작성할 줄 알면 풀리는 문제.
기본적으로 피보나치 수를 구할 때 재귀 + 메모이제이션(memoization)으로 구하였었는데 프로그래머스는 solution이라는 주어진 함수에 제출하는 것이다보니 재귀보다는 반복문으로 작성하여 풀었다.

def solution(n):
    answer = 0
    a = b = 1
    for i in range(n-2):
        answer = a + b
        a, b = b, answer
    return answer % 1234567

첫 제출은 위와 같이 작성하였다. n은 1이상이기 때문에 F(0) = 0을 고려하지 않았고 주어진 answer 변수는 return하는데 사용했다.

하지만, 아래처럼 더 간단하게 코드를 작성할 수 있다.

def solution(n):
    a, b = 0, 1
    for i in range(n):
        a, b = b, a + b
    return a % 1234567

F(0) = 0의 내용을 추가하면서 불필요한 변수와 단계를 줄이는... 지극히 파이썬스러운 코딩이다.