[JAVA] 백준 16198번: 에너지 모으기 ( 중급 1-10 )

문제

N개의 에너지 구슬이 일렬로 놓여져 있고, 에너지 구슬을 이용해서 에너지를 모으려고 한다.

i번째 에너지 구슬의 무게는 Wi이고, 에너지를 모으는 방법은 다음과 같으며, 반복해서 사용할 수 있다.

1. 에너지 구슬 하나를 고른다. 고른 에너지 구슬의 번호를 x라고 한다. 단, 첫 번째와 마지막 에너지 구슬은 고를 수 없다.
2. x번째 에너지 구슬을 제거한다.
3. Wx-1 × Wx+1의 에너지를 모을 수 있다.
4. N을 1 감소시키고, 에너지 구슬을 1번부터 N번까지로 다시 번호를 매긴다. 번호는 첫 구슬이 1번, 다음 구슬이 2번, ... 과 같이 매겨야 한다.

N과 에너지 구슬의 무게가 주어졌을 때, 모을 수 있는 에너지 양의 최댓값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 에너지 구슬의 개수 N(3 ≤ N ≤ 10)이 주어진다.

둘째 줄에는 에너지 구슬의 무게 W1, W2, ..., WN을 공백으로 구분해 주어진다. (1 ≤ Wi ≤ 1,000)

출력

첫째 줄에 모을 수 있는 에너지의 최댓값을 출력한다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.StringTokenizer;

public class baekjoon16198 {
    private static int n;
    private static int[] info;
    private static boolean[] visit;
    private static List<Integer> resultList;

    public static void main(String[] args) throws IOException {
//        4
//        1 2 3 4
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(br.readLine());
        visit = new boolean[n];
        info = new int[n];
        resultList = new ArrayList<>();

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            info[i] = Integer.parseInt(st.nextToken());
        }

        permutation16198(0, 0);
        Collections.sort(resultList);
        System.out.println(resultList.get(resultList.size() - 1));
    }

    private static void permutation16198(int turn, int sum) {
        if (turn == n - 2) {
            resultList.add(sum);
            return;
        }

        for (int i = 1; i < n - 1; i++) {
            if (!visit[i]) {
                visit[i] = true;
                int currentSum = calculateSum16198(i);
                permutation16198(turn + 1, sum + currentSum);
                visit[i] = false;
            }
        }
    }

    private static int calculateSum16198(int index) {
        int leftIndex = findLeft(index);
        int rightIndex = findRight(index);
        return info[leftIndex] * info[rightIndex];
    }

    private static int findRight(int index) {
        for (int i = index + 1; i < n; i++) {
            if (!visit[i]) {
                return i;
            }
        }
        throw new IllegalArgumentException("오른쪽 값이 존재하지 않습니다.");
    }

    private static int findLeft(int index) {
        for (int i = index - 1; i >= 0; i--) {
            if (!visit[i]) {
                return i;
            }
        }
        throw new IllegalArgumentException("왼쪽 값이 존재하지 않습니다.");
    }
}