[프로그래머스] 멀리 뛰기 #파이썬 #dp #level3 [연습문제]

https://programmers.co.kr/learn/courses/30/lessons/12914

코딩테스트 연습 - 멀리 뛰기

효진이는 멀리 뛰기를 연습하고 있습니다. 효진이는 한번에 1칸, 또는 2칸을 뛸 수 있습니다. 칸이 총 4개 있을 때, 효진이는
(1칸, 1칸, 1칸, 1칸)
(1칸, 2칸, 1칸)
(1칸, 1칸, 2칸)
(2칸, 1칸, 1칸)
(2칸, 2칸)
의 5가지 방법으로 맨 끝 칸에 도달할 수 있습니다. 멀리뛰기에 사용될 칸의 수 n이 주어질 때, 효진이가 끝에 도달하는 방법이 몇 가지인지 알아내, 여기에 1234567를 나눈 나머지를 리턴하는 함수, solution을 완성하세요. 예를 들어 4가 입력된다면, 5를 return하면 됩니다.

제한 사항

• n은 1 이상, 2000 이하인 정수입니다.

입출력 예

n	result
4	5
3	3

입출력 예 설명

입출력 예 #1
위에서 설명한 내용과 같습니다.

입출력 예 #2
(2칸, 1칸)
(1칸, 2칸)
(1칸, 1칸, 1칸)
총 3가지 방법으로 멀리 뛸 수 있습니다.

풀이

간단한 dp입니다. n번째로 뛰는 방법의 가짓수는 (n-1칸 뛴 후 1칸 뛰는 가짓수) + (n-2칸 뛴 후 2칸 뛰는 가짓수)입니다. 1 칸, 2 칸일 때의 값은 지정해주어야 하며, n이 1, 2일 때의 예외와 값을 % 1234567한 값을 리턴해야 한다는 점에 주의합니다.

def solution(n):
    if n < 2:
        return n
    
    dp = [0 for _ in range(n)]
    dp[0] = 1
    dp[1] = 2
    
    for i in range(2, n):
        dp[i] = dp[i-1] + dp[i-2]
        
    return dp[-1] % 1234567