ALGORITHM

[JAVA] 백준 16198번- 에너지 모으기

연듀 2023. 1. 4. 19:37

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

 

16198번: 에너지 모으기

N개의 에너지 구슬이 일렬로 놓여져 있고, 에너지 구슬을 이용해서 에너지를 모으려고 한다. i번째 에너지 구슬의 무게는 Wi이고, 에너지를 모으는 방법은 다음과 같으며, 반복해서 사용할 수 있

www.acmicpc.net

 

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

public class Main {
    static int n;
    static ArrayList<Integer> list = new ArrayList<>();
    static int max = Integer.MIN_VALUE;

    static void dfs(int cnt, int sum) {
        if (cnt == n - 2) { // 첫번째와 마지막 구슬 제외하고 다 뽑으면
            max = Math.max(max, sum);
            return;
        }
        for (int i = 1; i < list.size()-1; i++) {
            int target = list.get(i);
            int energy = list.get(i - 1) * list.get(i + 1);
            list.remove(i); // 리스트에서 해당 원소 삭제
            dfs(cnt + 1, sum + energy);
            list.add(i, target); // i번째 인덱스에 다시 넣음
        }
    }

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

        n = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            list.add(Integer.parseInt(st.nextToken()));
        }
        dfs(0, 0);

        System.out.println(max);
    }
}

'ALGORITHM' 카테고리의 다른 글

[JAVA] 백준 1987번- 알파벳  (0) 2023.01.06
[JAVA] 백준 2580번- 스도쿠  (0) 2023.01.06
[JAVA] 백준 9663번- N-Queen  (0) 2023.01.04
[JAVA] 백준 11663번- 선분 위의 점  (0) 2022.12.30
[JAVA] 백준 2512 - 예산  (0) 2022.12.30