개발/algorithm

[프로그래머스][level2] 피보나치 수 -python

zzi_on2 2021. 12. 27. 13:44

문제 설명

피보나치 수는 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, ... 와 같이 이어집니다.

문제링크

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) =

programmers.co.kr

 

풀이 

처음에 단순히 재귀함수로 풀었더니, 시간초과 

# 시간초과
def solution(n):
    answer = 0
    answer = fibo(n) % 1234567
    return answer

def fibo(n):
    if n == 0 or n == 1 :
        return n
    return fibo(n-1) + fibo(n-2)

이전에 공부했던 [이것이 취업을 위한 코딩테스트다 파이썬편]에서 다이나믹 프로그래밍 참고하여 풀었다 

다이나믹 프로그래밍이란,

  • 동일한 함수가 반복적으로 호출될 때
  • 큰 문제를 작게 나누고, 같은 문제라면 한 번씩만 풀어 문제를 효율적으로 해결하는 알고리즘 기법

재귀함수를 사용하는 탑다운 방식과 반복문을 사용하는 보텀업 방식이 있었는데, 

일반적으로 보텀업 방식이 오버헤드를 줄일 수 있다. 

def solution(n):
    d = [0] * n
    d[0] = 1
    d[1] = 1 
    for i in range(2, n):
    # 계산할 때마다 1234567로 나눈 나머지를 저장
        d[i] = (d[i-1] + d[i-2])%1234567
    return d[-1]