코딩테스트/프로그래머스

[프로그래머스] 멀리뛰기(Java/Python)

Jindory 2022. 4. 15. 22:13
반응형

문제 설명

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

전체코드(Java)

class Solution {
    public long solution(int n) {
        // n번까지 횟수를 저장할 dp배열을 선언한다.
        long dp[] = new long[n];
        // n번까지 순회하며 각 길이당 나올 수 있는 경우의 수를 dp에 저장한다.
        for(int i=0;i<n;i++){
            
            if(i<2){
                // i가 2보다 작을경우 i+1의 값을 저장한다.
                dp[i] = i+1;
            }else{
                // 멀리뛰기 판 길이가 3일 때
                // 1칸을 채우면 남은 길이가 2이므로 길이가 2일경우에 경우의 수를
                // 2칸을 채우면 남은 길이가 1이므로 길이가 1일경우에 경우의 수를 더하면 되므로
                // 자신의 index보다 한칸 앞에 2칸 앞의 값을 더하면 자신의 길이를 채우는 경우의 수이다.
                dp[i] = (dp[i-1] + dp[i-2])%1234567;
            }
        }
        return dp[n-1];
    }
}

전체코드(Python)

def solution(n):
    answer = []
    # n번까지 순회하며 각 길이당 나올 수 있는 경우의 수를 answer에 저장한다.
    for i in range(n):
        # 길이가 멀리뛰기 판 길이가 1일 경우 한가지, 
        # 멀리뛰기 판 길이가 2일 경우 2가지이므로 i+1을 저장한다.
        if i<=1:
            answer.append(i+1)
        else:
            # 멀리뛰기 판 길이가 3일 때
            # 1칸을 채우면 남은 길이가 2이므로 길이가 2일경우에 경우의 수를
            # 2칸을 채우면 남은 길이가 1이므로 길이가 1일경우에 경우의 수를 더하면 되므로
            # 자신의 index보다 한칸 앞에 2칸 앞의 값을 더하면 자신의 길이를 채우는 경우의 수이다.
            num = (answer[i-1]+answer[i-2])%1234567
            answer.append(num)
    
    return answer[-1]

[참고]

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

반응형