[코딩 문제 풀이(파이썬/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이하인 자연수입니다.

 

재귀함수로 구성하게 되면, 시간초과에 걸린다.

# 시간 초과
def solution(n):
    if n==0 :
        return 0
    elif n==1:
        return 1
    elif n > 1 :
        return (solution(n-1) + solution(n-2))%1234567

두 개를 더하는 리스트를 만드는 것으로 해결했다.

 

def solution(n):
    answer = [0,1]
    for i in range(n-1):
        answer.append(answer[i] +answer[i+1])
    return answer[-1]%1234567

메모리, 효율성 측면에서 훨씬 우수한 것은 숫자를 두 개씩만 다루는 것이다.

def solution(num):
    a,b = 0,1
    for i in range(num):
        a,b = b,a+b
        # a = b, b = a+b로 쓰지 않도록 유의하자.
    return a%1234567

a,b = b, a+b 라고 한 줄에 적어주어야 동시에 계산되어 원하는대로 계산된다.