[BOJ-#2748] 피보나치 수 2

문제

문제 링크
피보나치 수는 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은 90보다 작거나 같은 자연수이다.

출력

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

---

Code

Github 링크

import sys

input = sys.stdin.readline

test_case = int(input().rstrip())

num1 , num2 = 0, 1
answer = 0 

for i in range(test_case) :
    if i == 0 :
        pass 
    else :
        answer = num1 + num2
        num1 = num2
        num2 = answer


if test_case == 1 :
    print(1)
else :
    print(answer)

코드 설명

import sys

input = sys.stdin.readline

test_case = int(input().rstrip())

num1 , num2 = 0, 1
answer = 0 

빠른 입력을 받을 수 있도록 input을 선언하고 테스트 케이스의 개수를 받는다. 0번째 피보나치 수가 0으로 시작하므로 num1과 num2를 0, 1로 설정하고, answer을 0으로 선언한다.

for i in range(test_case) :
    if i == 0 :
        pass 
    else :
        answer = num1 + num2
        num1 = num2
        num2 = answer


if test_case == 1 :
    print(1)
else :
    print(answer)

for문을 돌려서 i가 0일 경우를 제외하고 num1과 num2를 더한 뒤, num2를 num1로 바꾸고 num2를 answer으로 바꾼다. 피보나치 수열으로 보자면 한 칸씩 앞으로 당기는 과정이다.

이때 test_case가 1일 경우에는 for문이 동작하지 않으므로 1일 경우에만 1을 출력해 주는 if문을 추가해 준다.

---

리뷰

DP의 근본,,, 피보나치 수열을 풀어보았다. 브론즈라 그런지 빨리 풀어서 다음에는 피보나치 수가 5까지 있어서 피보나치 시리즈를 풀어볼 예정이다.