[코딩 테스트 연습(파이썬/Python)] 백준 10870번 _ 피보나치 수 1~5

https://www.acmicpc.net/problem/10870

10870번: 피보나치 수 5

[문제]

피보나치 수는 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번째 피보나치 수를 구하는 프로그램을 작성하시오.

 

[입력]

첫째 줄에 n이 주어진다. n은 20보다 작거나 같은 자연수 또는 0이다.

[출력]

첫째 줄에 n번째 피보나치 수를 출력한다.

 

[아이디어]

1) 재귀 함수를 이용하자

2) 두 개의 항씩만 저장하는 dp를 만들자

 

n=int(input())
def f(n):
    if n==0: return 0
    elif n==1: return 1
    else: return f(n-1)+f(n-2)
print(f(n))

위처럼 풀면 조건에 따라 금방 시간초과 혹은 메모리 초과 혹은 파이썬의 기본 재귀 깊이를 초과하게 될 것이다.

# dp로
n=int(input())
dp = [0,1]
for _ in range(n-1):
    dp = [dp[1],dp[0]+dp[1]]
print(dp[0] if n==0 else dp[1])

dp를 이용해 풀면 위와 같이 해결할 수 있다.