본문 바로가기

Tech/Problem Solving

[백준 2579] 계단 오르기 (Java)

https://www.acmicpc.net/problem/2579

 

2579번: 계단 오르기

계단 오르기 게임은 계단 아래 시작점부터 계단 꼭대기에 위치한 도착점까지 가는 게임이다. <그림 1>과 같이 각각의 계단에는 일정한 점수가 쓰여 있는데 계단을 밟으면 그 계단에 쓰여 있는 점수를 얻게 된다. 예를 들어 <그림 2>와 같이 시작점에서부터 첫 번째, 두 번째, 네 번째, 여섯 번째 계단을 밟아 도착점에 도달하면 총 점수는 10 + 20 + 25 + 20 = 75점이 된다. 계단 오르는 데는 다음과 같은 규칙이 있다. 계단은 한 번에 한 계단씩

www.acmicpc.net

접근 방법

 

DP 방법으로 풀었으며, N번째 계단까지 올라올 때의 경우는 두가지가 있다.

 

첫 번째는 n-1 번째를 밟지 않고 온 경우

두 번째는 n-1 번째를 밟고 온 경우

 

각각의 점화식은 다음과 같으며 이 중 최대값을 골라 dp[] 배열에 넣으면 된다.

 

 

소스 코드

import java.util.Scanner;

public class Main {
    private static int[] stairs;

        public static void main(String[] args) {
            Scanner scanner = new Scanner(System.in);

            int N = scanner.nextInt();
            stairs = new int[N + 1];

            for (int i = 1; i <= N; i++) {
                stairs[i] = scanner.nextInt();
            }

            int[] dp = new int[N + 1];

            dp[1] = stairs[1];

            if (N >= 2) {
                dp[2] = dp[1] + stairs[2];
            }

            for (int i = 3; i <= N; i++) {
                dp[i] = Math.max(dp[i - 3] + stairs[i] + stairs[i - 1], dp[i - 2] + stairs[i]);
            }

            System.out.println(dp[N]);
        }
}

 

 

 

반응형