본문 바로가기
백준알고리즘/동적 계획법1

(Python/🥈4)백준리즘알고리즘 15624번: 피보나치 수 7

by windy7271 2023. 7. 24.
728x90
반응형

 

문제 바로가기 

 

문제:

피보나치 수는 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은 1,000,000보다 작거나 같은 자연수 또는 0이다.

 

출력:

첫째 줄에 n번째 피보나치 수를 1,000,000,007으로 나눈 나머지를 출력한다.

 

풀이:

 

틀린 풀이

import sys
sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

N = int(input())
dp = [0] * (N+1)

dp[1] =1 ; dp[2] = 1
for i in range(3,N+1):
    dp[i] = dp[i-1] + dp[i-2]
print(dp[N]%1000000007)

기본적인dp 로 풀면 메모리 초과가 나온다.

 

그래서 아예 나머지 값을 넣기로 해보았다.

 

import sys
sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

N = int(input())
dp = [0] * (max(N + 1, 3))

dp[1] = 1
dp[2] = 1

for i in range(3, N + 1):
    dp[i] = (dp[i - 1] + dp[i - 2]) % 1000000007

print(dp[N])

 

그냥 해주면 3 이하는 IndexError 가 나기 때문에 에러처리해준다

반응형

댓글