백준 1003: 피보나치 함수

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

문제

다음 소스는 N번째 피보나치 수를 구하는 C++ 함수이다.

int fibonacci(int n) {
    if (n == 0) {
        printf("0");
        return 0;
    } else if (n == 1) {
        printf("1");
        return 1;
    } else {
        return fibonacci(n‐1) + fibonacci(n‐2);
    }
}

fibonacci(3)을 호출하면 다음과 같은 일이 일어난다.

• fibonacci(3)은 fibonacci(2)와 fibonacci(1) (첫 번째 호출)을 호출한다.
• fibonacci(2)는 fibonacci(1) (두 번째 호출)과 fibonacci(0)을 호출한다.
• 두 번째 호출한 fibonacci(1)은 1을 출력하고 1을 리턴한다.
• fibonacci(0)은 0을 출력하고, 0을 리턴한다.
• fibonacci(2)는 fibonacci(1)과 fibonacci(0)의 결과를 얻고, 1을 리턴한다.
• 첫 번째 호출한 fibonacci(1)은 1을 출력하고, 1을 리턴한다.
• fibonacci(3)은 fibonacci(2)와 fibonacci(1)의 결과를 얻고, 2를 리턴한다.

1은 2번 출력되고, 0은 1번 출력된다. N이 주어졌을 때, fibonacci(N)을 호출했을 때, 0과 1이 각각 몇 번 출력되는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다.

각 테스트 케이스는 한 줄로 이루어져 있고, N이 주어진다. N은 40보다 작거나 같은 자연수 또는 0이다.

출력

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

---

아이디어

우선, N이 3일 경우 정답은 N이 1일때와 2일때의 0과 1의 출력 수를 각각 합한 것이다.

제약사항으로 N은 40이하라고 하였으니 0, 1일때 초기값을 넣어주고 2부터 40까지는 반복문을 한번 돌려서 미리 구해놓으면 그 다음부터는 출력만 하면 끝이다!

이렇게 하면 시간이 0.25초라도 아주 널널하다!

제출 코드

# https://www.acmicpc.net/problem/1003
# 아이디어 : 어차피 N이 40이 최대이므로 0~40까지 미리 다 구해놓는다.
# 정답의 원리는 n-1, n-2에 출력된 개수의 합이므로 한번 쭉 만들어놓는다면
# 시간은 O(N)이다. 시간이 0.25초라도 아주 널널하다!

fibo_list = [[1, 0], [0, 1]]


for i in range(2, 41):
    fibo_list.append(
        [fibo_list[-1][0] + fibo_list[-2][0], fibo_list[-1][1] + fibo_list[-2][1]]
    )

TC = int(input())
for tc in range(TC):
    n = int(input())
    print(fibo_list[n][0], fibo_list[n][1])​