Problem Solving/Baekjoon

[백준] 2579 계단 오르기 - Dynamic Programming / Java

graycode 2022. 10. 19. 23:17

• 문제 링크

 

2579번: 계단 오르기

계단 오르기 게임은 계단 아래 시작점부터 계단 꼭대기에 위치한 도착점까지 가는 게임이다. <그림 1>과 같이 각각의 계단에는 일정한 점수가 쓰여 있는데 계단을 밟으면 그 계단에 쓰여 있는 점

www.acmicpc.net

 

• 풀이 과정

 

• 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(br.readLine());

        int[] arr = new int[n + 1];
        int[] dp = new int[n + 1];

        for (int i = 1; i <= n; i++)
            arr[i] = Integer.parseInt(br.readLine());

        dp[1] = arr[1];

        for (int i = 2; i <= n; i++) {
            if (i == 2)
                dp[2] = arr[1] + arr[2];
            else if (i == 3)
                dp[3] = Math.max(arr[1], arr[2]) + arr[3];
            else
                dp[i] = Math.max(dp[i - 3] + arr[i - 1], dp[i - 2]) + arr[i];
        }

        bw.write(String.valueOf(dp[n]));
        bw.flush();
    }

}