프로그래머스 12945번 피보나치 수 - Python

문제 설명

프로그래머스 12945번 피보나치 수

피보나치 수는 F(0) = 0, F(1) = 1일 때, 1 이상의 n에 대하여 F(n) = F(n-1) + F(n-2) 가 적용되는 수 입니다. 예를들어

• F(2) = F(0) + F(1) = 0 + 1 = 1
• F(3) = F(1) + F(2) = 1 + 1 = 2
• F(4) = F(2) + F(3) = 1 + 2 = 3
• F(5) = F(3) + F(4) = 2 + 3 = 5

와 같이 이어집니다.

2 이상의 n이 입력되었을 때, n번째 피보나치 수를 1234567으로 나눈 나머지를 리턴하는 함수, solution을 완성해 주세요.

제한 사항

• n은 2 이상 100,000 이하인 자연수입니다.

입출력 예시

n	return
3	2
5	5

입출력 예 설명 1, 2

• 피보나치수는 0번째부터 0, 1, 1, 2, 3, 5, … 와 같이 이어집니다.

코드 구현

Python

def solution(n):
    answer = [0, 1]

    for i in range(2, n + 1):
        answer.append(answer[i - 2] + answer[i - 1])

    return answer[n] % 1234567

시간 복잡도 - Python

• 반복문을 실행하는데 시간 복잡도는 $O(n)$이다.
• 따라서, 시간 복잡도는 $O(n)$이다.

개선할 점 - Python

• 배열 대신에 변수를 사용하면 시간 복잡도는 그대로 이지만 공간 복잡도를 매우 작게 할 수 있다.

def solution(n):
    a, b = 0, 1

    for i in range(2, n + 1):
        a, b = b, a + b

    return b % 1234567