[Python] 파이썬 백준(24416) 피보나치 수1

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

24416번: 알고리즘 수업 - 피보나치 수 1

---

나의 풀이

오늘도 서준이는 동적 프로그래밍 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해 보자.

오늘은 n의 피보나치 수를 재귀호출과 동적 프로그래밍으로 구하는 알고리즘을 배웠다. 재귀호출에 비해 동적 프로그래밍이 얼마나 빠른지 확인해 보자. 아래 의사 코드를 이용하여 n의 피보나치 수를 구할 경우 코드 1 코드 2 실행 횟수를 출력하자.

위의 코드를 파이썬으로 작성하여 문제 해결

import sys

n = int(sys.stdin.readline())

def refib(n, cnt = 0):
    if n==1 or n==2:
        return 1
    else:
        return refib(n-1)+refib(n-2)

def dpfib(n, cnt = 0):
    f=[1,1]
    if n == 1 or n == 2:
        return f[n-1]
    for i in range(1, n-1):
        cnt += 1
        f.append(f[i-1]+f[i])
    return cnt

print(refib(n))
print(dpfib(n,0))

---

다른 사람의 풀이

다른 사람들은 dp 문제 풀 때 dp=[0]*(n+1) 이렇게 선언해 주는데 난 append를 사용한다. 

둘 다 비슷한데 미리 선언해 주는 게 더 깔끔해 보이니 앞으로 선언해서 써야겠다.

import sys
def fib(n):
    if n==1 or n==2:
        return 1
    else:
        return fib(n-1)+fib(n-2)
        
def fibonacci(n):
    dp=[0]*(n+1)
    dp[1],dp[2]=1,1
    cnt2=0
    for i in range(3,n+1):
        cnt2+=1
        dp[i]=dp[i-1]+dp[i-2]
    return cnt2

n = int(sys.stdin.readline())
print(fib(n),fibonacci(n))

 

• 재귀로 인해 python으로 제출하면 시간초과가 뜬다 pypy3으로 하자..!

---

Python으로 제출하고 싶다면?

import sys

n = int(sys.stdin.readline())

def dpfib(n, cnt = 0):
    f=[1,1]
    if n == 1 or n == 2:
        return f[n-1]
    for i in range(1, n-1):
        cnt += 1
        f.append(f[i-1]+f[i])
    return f[-1],cnt

f, cnt = dpfib(n,0)
print(f, cnt, sep='\n')