에너지 모으기

Huisu·2024년 1월 4일
0

Coding Test Practice

목록 보기
89/119
post-thumbnail

문제

16198번: 에너지 모으기

문제 설명

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

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

  1. 에너지 구슬 하나를 고른다. 고른 에너지 구슬의 번호를 x라고 한다. 단, 첫 번째와 마지막 에너지 구슬은 고를 수 없다.

  2. x번째 에너지 구슬을 제거한다.

  3. W × W의 에너지를 모을 수 있다.

    x-1

    x+1

  4. N을 1 감소시키고, 에너지 구슬을 1번부터 N번까지로 다시 번호를 매긴다. 번호는 첫 구슬이 1번, 다음 구슬이 2번, ... 과 같이 매겨야 한다.

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

제한 사항

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

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

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

입출력 예

입력출력
4
1 2 3 412
5
100 2 1 3 10010400
7
2 2 7 6 90 5 91818
10
1 1 1 1 1 1 1 1 1 18

아이디어

dfs

제출 코드

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class one16198 {
    public int n;
    public int max = Integer.MIN_VALUE;

    public void solution() throws IOException {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        List<Integer> weighted = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            weighted.add(sc.nextInt());
        }

        dfs(weighted, 0);

        System.out.println(max);
    }

    private void dfs(List<Integer> weighted, int sum) {
        if (weighted.size() <= 2) {
            max = Math.max(max, sum);
        }

        for (int i = 1; i < weighted.size() - 1; i++) {
            int temp = weighted.get(i);
            int addEnergy = (weighted.get(i - 1) * weighted.get(i + 1));
            weighted.remove(i);
            dfs(weighted, sum + addEnergy);
            weighted.add(i, temp);
        }
    }

    public static void main(String[] args) throws IOException {
        new one16198().solution();
    }
}

0개의 댓글