[백준 / 실버1] 2156 포도주 시식 (Java)

문제 보기

사용한 것

• 점화식을 세워 풀이하기 위한 bottom-up

풀이 방법

• dp[i][0] : 이번꺼 선택 X
  • dp[i - 1][j] 중 가장 큰 값
• dp[i][1] : 이번꺼 처음으로 선택
  • dp[i - 2][j] 중 가장 큰 값 + arr[i]
• dp[i][2] : 이번꺼 두번째로 선택
  • dp[i - 1][1] + arr[i]

코드

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[] arr = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            arr[i] = Integer.parseInt(br.readLine());
        }
        int[][] dp = new int[n + 1][3];
        dp[1][0] = 0;
        dp[1][1] = arr[1];
        if(n > 1) {
            dp[2][0] = arr[1];
            dp[2][1] = arr[2];
            dp[2][2] = arr[1] + arr[2];
        }
        for(int i = 3; i <= n; i++) {
            for(int j = 0; j < 3; j++) {
                dp[i][0] = Math.max(dp[i - 1][j], dp[i][0]);
            }
            for(int j = 0; j < 3; j++) {
                dp[i][1] = Math.max(dp[i - 2][j] + arr[i], dp[i][1]);
            }
            dp[i][2] = dp[i - 1][1] + arr[i];
        }

        int answer = 0;
        for(int i = 0; i < 3; i++) {
            answer = Math.max(dp[n][i], answer);
        }
        System.out.println(answer);
    }
}