[프로그래머스/Python] 피보나치 수

문제 정보

제목 : 피보나치 수

난이도 : Lv.2

사용 언어 : Python

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/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

입출력 예 설명

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

---

나의 풀이

a와 b에 초기값으로 0,1을 저장합니다. n까지 반복하는 for문을 생성하고 파이썬으로 가능한 문법을 이용하여 a에는 b를 저장하고 b에는 a+b를 저장해줍니다. 반복이 끝나면 answer 변수에 a를 1234567로 나눈 나머지를 구하여 리턴합니다.

 

코드

def solution(n):
    answer = 0
    a = 0
    b = 1 
    
    for i in range(n):
        a,b = b,a+b
        
    answer = a % 1234567
    
    return answer