[파이썬] 백준 24416번 알고리즘 수업

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

시간 제한

 1초

메모리 제한

512MB

문제

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

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

피보나치 수 재귀호출 의사 코드는 다음과 같다.

fib(n) {
    if (n = 1 or n = 2)
    then return 1;  # 코드1
    else return (fib(n - 1) + fib(n - 2));
}

피보나치 수 동적 프로그래밍 의사 코드는 다음과 같다.

fibonacci(n) {
    f[1] <- f[2] <- 1;
    for i <- 3 to n
        f[i] <- f[i - 1] + f[i - 2];  # 코드2
    return f[n];
}

입력

첫째 줄에 n(5 ≤ n ≤ 40)이 주어진다.

출력

코드1 코드2 실행 횟수를 한 줄에 출력한다.

예제 입력 1 

5

예제 출력 1 

5 3

예제 입력 2 

30

예제 출력 2 

832040 28

---

나의 풀이

n = 3 부터 반복되기 때문에 1, 2는 미리 빼뒀다.

def fib(n):
    global count_fib
    count_fib += 1
    if n == 1 or n == 2:
        count_fib -= 1
        return 1
    else:
        return (fib(n - 1) + fib(n - 2))

dp = {1 : 1, 2 : 1}
def fib_dp(n):
    global count_dp
    # n이 dp table에 있는 경우
    if n in dp:
        return dp[n]
    # n이 dp table에 없는 경우
    else:
        dp[n] = fib_dp(n - 1) + fib_dp(n  - 2)
        count_dp += 1
        return dp[n]
    
n = int(input())
count_fib, count_dp = 0, 0
fib(n)
fib_dp(n)
# count_fib에 1, 2
print(count_fib + 1, count_dp)

다른 사람 풀이

의사 코드를 그대로 구현한 풀이다. 범위가 정해져있기 때문에 list로도 풀수 있다.

def fib(n):
    global cnt1
    cnt1+=1
    if n == 1 or n == 2:
        cnt1-=1
        return 1;
    else:
        return fib(n-1) + fib(n-2)
    
def fibb(n):
    global cnt2
    f[1],f[2]=1,1
    for i in range(3,n+1):
        cnt2+=1
        f[i] = f[i-1]+f[i-2]
    return f[n]

f = [0 for _ in range(41)]
n = int(input())
cnt1,cnt2=0,0
fib(n)
fibb(n)
print(cnt1+1,cnt2)

[BOJ][Python]24416번 풀이