코딩테스트/[백준] 코딩테스트 연습
에너지 모으기 - 16198번
쵼쥬
2021. 10. 8. 14:54
내 코드
package com.company;
import java.io.*;
import java.util.*;
public class Main {
static boolean[] visited;
static int N;
static ArrayList<Integer> W;
static int max = 0;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = null;
N = Integer.parseInt(br.readLine());
W = new ArrayList<>();
st = new StringTokenizer(br.readLine());
for (int i = 0; i < N; i++) {
W.add(Integer.parseInt(st.nextToken()));
}
visited = new boolean[N];
visited[0] = true;
visited[N - 1] = true;
for (int i = 1; i < N - 1; i++) {
dfs(i, 0);
}
if (N == 3)
max = W.get(0) * W.get(2);
System.out.println(max);
}
static void dfs(int x, int sum) {
if (visited[x]) {
max = Math.max(sum, max);
return;
}
visited[x] = true;
int w1 = W.get(0);
int w2 = W.get(N - 1);
int z = 1;
while (x - z > 0) {
if (!visited[x - z]) {
w1 = W.get(x - z);
break;
}
z++;
}
z = 1;
while (x + z < N - 1) {
if (!visited[x + z]) {
w2 = W.get(x + z);
break;
}
z++;
}
sum += w1 * w2;
for (int i = 1; i < W.size() - 1; i++) {
if (i == x)
continue;
dfs(i, sum);
}
visited[x] = false;
}
}