[백준 알고리즘] 2156번 포도주 시식 (JAVA)

 

---

주의 사항

• 포도주를 마시는 경우의 수가 3가지 이기 때문에 i = 1, i = 2, i = 3을 계산 후 i = 4 일때부터 시작해야한다.
  • OOX, OXO, XOO (총 3가지)

 

---

코드 블럭

package boj401;

import java.util.Scanner;

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

        int n = scanner.nextInt();

        int[] arr = new int[n];

        for (int i = 0; i < n; i++) {
            arr[i] = scanner.nextInt();
        }

        int[] dp = new int[n];

        if (n < 3) {
            if (n == 1) {
                System.out.println(arr[0]);
                return;
            } else if (n == 2) {
                System.out.println(arr[0] + arr[1]);
                return;
            } else if (n == 3) {
                System.out.println(Math.max(dp[1], Math.max(arr[0] + arr[2], arr[1] + arr[2])));
                return;
            }
        }
        dp[0] = arr[0];
        dp[1] = arr[0] + arr[1];
        dp[2] = Math.max(dp[1], Math.max(arr[0] + arr[2], arr[1] + arr[2]));

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

        System.out.println(dp[n - 1]);
    }

}