본문 바로가기

취준/백준

[백준][Python][2747][lru_cache] 피보나치 수

문제 : https://www.acmicpc.net/problem/2747

 

2747번: 피보나치 수

피보나치 수는 0과 1로 시작한다. 0번째 피보나치 수는 0이고, 1번째 피보나치 수는 1이다. 그 다음 2번째 부터는 바로 앞 두 피보나치 수의 합이 된다. 이를 식으로 써보면 Fn = Fn-1 + Fn-2 (n>=2)가 된다. n=17일때 까지 피보나치 수를 써보면 다음과 같다. 0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233, 377, 610, 987, 1597 n이 주어졌을 때, n번째 피보나치 수를 구하는

www.acmicpc.net

python 3.2 이상 기준이다.

functools 의 lru_cache decorator 는 memoization을 자동으로 한다.

지금 밑의 코드를 보면 재귀로 피보나치를 구현했다. 하지만, 파이썬 인터프리터가 자동으로 함수의 값을 저장해 memoization과 같은 방식으로 동작하게 한다.

 

import functools


@functools.lru_cache()
def fibo(n):
    if n == 0:
        return 0
    elif n == 1:
        return 1
    else:
        return fibo(n-1)+fibo(n-2)


print(fibo(int(input())))