로그인회원가입
[코딩테스트] 프로그래머스 Lv2. 멀리뛰기(DP)

2026.01.15. 14:57

문제 설명

효진이는 멀리 뛰기를 연습하고 있습니다. 효진이는 한번에 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 이하인 정수입니다.

입출력 예

nresult4533

입출력 예 설명

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

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

처음엔 DFS로 풀었는데 시간초과가 뜬다.

def solution(n):
    answer = []
    way = []
    dfs(n, way, answer)
    
    return len(answer)%1234567

def dfs(n, way, answer):
    
    if n == sum(way): # 전체 들어간 값이 n이면 리턴
        answer.append(way.copy()) #way 참조 문제로 복사값을 넣어야함.
        return;
    elif n < sum(way): #더 멀리 뛴 경우
        return
    
    # 1아니면 2를 넣고 pop()
    for i in [1,2] : 
        way.append(i)
        dfs(n, way, answer)
        way.pop()

DP(동적계획법)로 풀어야 된다고 함.

피보나치 수열 DP 로 풀기

점화식 

n = 1  -  1

n = 2  -  1,1  /  2

n = 3  - 1,1,1 / 2,1  / 1,2   ( n = 2 에다가 뒤에 1을 붙인 경우 + n =1 에다가 뒤에 2를 붙인 경우 )

n = 4  - 1,1,1,1 / 2,1,1 / 1,2,1 / 1,1,2 / 2,2   (n = 3 에다가 뒤에 1을 붙인 경우 + n = 2에다가 뒤에 2를 붙인 경우)

즉 n = n-1에 1을 붙인 경우의 수  + n-2에 2를 뒤에 붙인 경우의 수

n의 개수 = n-1의 개수 + n-2의 개수

def solution(n):
    if n == 1 :
        return 1
    if n == 2:
        return 2
    
    arr = [0]*(n+1)
    arr[1] = 1 
    arr[2] = 2
        
    for i in range(3, n+1):
        arr[i] = (arr[i-1]+arr[i-2])
    
    return arr[n]%1234567

테스트 11 〉	통과 (0.36ms, 9.21MB)
테스트 12 〉	통과 (0.33ms, 9.34MB)
테스트 13 〉	통과 (0.31ms, 9.23MB)
테스트 14 〉	통과 (0.61ms, 9.23MB)
테스트 15 〉	통과 (0.51ms, 9.09MB)
테스트 16 〉	통과 (0.38ms, 9.29MB)
def solution(n):
    if n == 1 :
        return 1
    if n == 2:
        return 2
    
    arr = [0]*(n+1)
    arr[1] = 1 
    arr[2] = 2
        
    for i in range(3, n+1):
        arr[i] = (arr[i-1]+arr[i-2])%1234567
    
    return arr[n]


테스트 11 〉	통과 (0.31ms, 9.25MB)
테스트 12 〉	통과 (0.49ms, 9.07MB)
테스트 13 〉	통과 (0.31ms, 9.23MB)
테스트 14 〉	통과 (0.36ms, 9.26MB)
테스트 15 〉	통과 (0.25ms, 9.18MB)
테스트 16 〉	통과 (0.32ms, 9.16MB)

DP로 풀었을 때 결과. 

나머지 연산 위치에 따라 시간이 약간 달라진다.

목록