boj16198 에너지 모으기

dgh03207·2022년 5월 6일
0

알고리즘

목록 보기
33/45

링크

https://www.acmicpc.net/problem/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)

출력

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

나의 풀이

  • dfs방식으로 풀이하고, 배열에서 고른 부분은 제거하기로 하였다.그리고 만약 배열의 크기가 3이되면 1 인덱스에 놓인 것을 제거했을때의 결과를 마지막으로 계산하고, return 해준다.
  • 핵심 코드
        private static void dfs(ArrayList<Integer> marbles, int weight){
            if(marbles.size()==3){
                weight += marbles.get(0)*marbles.get(2);
                if(answer< weight) answer = weight;
                return;
            }
    
            for (int i = 1; i < marbles.size()-1; i++) {
                ArrayList<Integer> temp = (ArrayList<Integer>)marbles.clone();
                temp.remove(i);
                dfs(temp,weight+(temp.get(i-1)*temp.get(i)));
            }
        }
  • 전체 코드
    package Baekjoon.java.silver;
    
    import java.io.BufferedReader;
    import java.io.InputStreamReader;
    import java.util.ArrayList;
    import java.util.List;
    import java.util.StringTokenizer;
    
    public class boj16198 {
        static int answer =0;
        public static void main(String[] args) throws Exception{
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            int N = Integer.parseInt(br.readLine());
            ArrayList<Integer> marbles = new ArrayList<>();
            StringTokenizer st = new StringTokenizer(br.readLine());
            for (int i = 0; i < N; i++) {
                marbles.add(Integer.parseInt(st.nextToken()));
            }
            dfs(marbles,0);
            System.out.println(answer);
    
        }
    
        private static void dfs(ArrayList<Integer> marbles, int weight){
            if(marbles.size()==3){
                weight += marbles.get(0)*marbles.get(2);
                if(answer< weight) answer = weight;
                return;
            }
    
            for (int i = 1; i < marbles.size()-1; i++) {
                ArrayList<Integer> temp = (ArrayList<Integer>)marbles.clone();
                temp.remove(i);
                dfs(temp,weight+(temp.get(i-1)*temp.get(i)));
            }
        }
    
    }

결과

profile
같이 공부하자!

0개의 댓글