[프로그래머스] DP - 멀리 뛰기 / Python
LEVEL 3 - ‘멀리 뛰기’ 문제 보러 가기 !
문제 설명
효진이는 멀리 뛰기를 연습하고 있습니다.
효진이는 한번에 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
과 같이 변수의 조건의 범위가 크거나 1234567
와 같이 결과값이 커서 나머지를 출력하라는 문제 거의 대부분은 DP 문제
입니다.
일반적인 완전 탐색의 접근법으로는 시간 초과가 나서 테스트 케이스를 전부 통과할 수 없습니다.
따라서, 멀리 뛰기
문제도 DP로 접근했고 주어진 테스트 케이스에 대해서 규칙을 찾아보았습니다.
- n=1일 때, 1가지
- n=2일 때, 2가지
- n=3일 때, 3가지
- n=4일 때, 5가지
규칙이 무엇인지 보이시나요?!
이 문제는 피보나치 수열
을 구하는 문제와 동일합니다.
점화식으로 표현하면 다음과 같습니다.
DP(N) = DP(N-1) + DP(N-2)
풀이 코드
Python
def solution(n):
memo = [-1 for _ in range(2001)]
def dp(x):
if memo[x] != -1: return memo[x]
if x == 1: return 1
if x == 2: return 2
memo[x] = (dp(x-2) + dp(x-1)) % 1234567
return memo[x]
return dp(n)