백준알고리즘 2156번 포도주 시식

백준알고리즘

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

풀이는 http://zoonvivor.tistory.com/133를 참고하자

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int count = Integer.parseInt(scan.nextLine().trim());
        int inputArray[] = new int[count];
        int memoArray[] = new int[count];
        
        for(int i=0; i < inputArray.length; i++) {
            inputArray[i] = Integer.parseInt(scan.nextLine().trim());
        }
        
        if(count >= 1) {
            memoArray[0] = inputArray[0];
        }
        
        if(count >= 2) {
            memoArray[1] = inputArray[0] + inputArray[1];
        }
        
        if(count >= 3) {
            memoArray[2] = 
            Math.max(memoArray[1], Math.max(inputArray[2] + inputArray[1], inputArray[2] + inputArray[0]));
        }
        
        for(int i=3; i<inputArray.length; i++) {
            memoArray[i] = 
            Math.max(memoArray[i-1], 
            Math.max(inputArray[i] + memoArray[i-2], inputArray[i] + inputArray[i-1] + memoArray[i-3]));
        }
        
        System.out.println(memoArray[count-1]);
        
        scan.close();
    }
}