[ Programmers / Coding Test / Python ] 멀리 뛰기

황승환·2022년 4월 21일
0

Python

목록 보기
274/498

문제 설명

효진이는 멀리 뛰기를 연습하고 있습니다. 효진이는 한번에 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가지 방법으로 멀리 뛸 수 있습니다.

접근 방법

이 문제와 같이 n이 몇일 때의 경우의 수를 구하는 문제는 DP를 통해 쉽게 해결할 수 있다. 그리고 DP 문제의 경우 매우 큰 수를 나눈 나머지값을 구하라는 조건이 보통 주어지기 때문에 이를 통해서도 DP를 사용해야 함을 알 수 있다. 점화식을 구해보았는데 생각보다 너무 간단한 점화식이 구해졌다.
dp[i]=dp[i-1]+dp[i-2]
이 점화식을 통해 코드를 구현하였다.

solution.py

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

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글