본문 바로가기

알고리즘/python

[python/파이썬] 백준 1003 피보나치 함수

반응형

[문제 출처]

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

 

1003번: 피보나치 함수

각 테스트 케이스마다 0이 출력되는 횟수와 1이 출력되는 횟수를 공백으로 구분해서 출력한다.

www.acmicpc.net

 

 

[문제 풀이]

피보나치 수의 원리만 그대로 이용하면 어렵지 않게 풀 수 있는 문제이다.

하지만 입력이 0부터 가능하기 때문에 0 혹은 1일 때와 2 이상의 숫자일 때를 잘 구분하여 코드를 작성해야 한다.

import sys
input = sys.stdin.readline

T = int(input())

for _ in range(T):
    n = int(input())

    dp = [[0, 0] for _ in range(n+1)]
    dp[0] = [1,0]
    if n>0:
        dp[1] = [0,1]

    if n>1:
        for i in range(2, n+1):
            dp[i] = [dp[i-1][0] + dp[i-2][0], dp[i-1][1] + dp[i-2][1]]

    print(dp[n][0],dp[n][1])
반응형