[프로그래머스] LV2. 멀리 뛰기 (파이썬, Python)

 

문제 링크

 

구분

코딩테스트 연습 > 연습문제

 

문제 설명

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

 

 

 

첫 풀이

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

뭘 틀린 걸까?

n을 알려면 n+1까지 진행해야한다. => dp크기 정의, 반복문 범위, answer 수정 요망

만약 n = 1이라면 dp = [0]이 된다. 따라서 dp[1], dp[2]에 에러가 난다. => 조건문 추가하기

 

 

 

 

 

문제 풀이

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