


내 코드
package com.company;
import java.io.*;
import java.util.*;
public class Main {
static int N, answer;
static int[][] arr;
static int[] p;
static boolean[] visited;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = null;
N = Integer.parseInt(br.readLine());
p = new int[N + 1];
arr = new int[N + 1][N + 1];
visited = new boolean[N + 1];
answer = -1;
st = new StringTokenizer(br.readLine());
for (int i = 1; i <= N; i++) {
p[i] = Integer.parseInt(st.nextToken());
}
for (int i = 1; i <= N; i++) {
st = new StringTokenizer(br.readLine());
int x = Integer.parseInt(st.nextToken());
for (int j = 0; j < x; j++) {
int y = Integer.parseInt(st.nextToken());
arr[i][y] = 1;
arr[y][i] = 1;
}
}
dfs(1, 0);
System.out.println(answer);
}
static void dfs(int index, int x) {
int a = bfs(x);
int b = bfs((int) Math.pow(2, N) - 1 - x);
if (a != -1 && b != -1 && (answer == -1 || answer > Math.abs(a - b))) {
answer = Math.abs(a - b);
}
for (int i = index; i <= N; i++) {
dfs(i + 1, x + (1 << i - 1));
dfs(i + 1, x);
}
}
static int bfs(int x) {
int index = 0;
for (int i = 1; i <= N; i++) {
if ((1 << (i - 1) & x) > 0) {
index = i;
break;
}
}
Queue<Integer> q = new LinkedList<>();
q.add(index);
x -= 1 << (index - 1);
int total = 0;
while (!q.isEmpty()) {
int a = q.poll();
total += p[a];
for (int i = 1; i <= N; i++) {
if (arr[a][i] == 1 && (x & 1 << (i - 1)) > 0) {
x -= 1 << (i - 1);
q.add(i);
}
}
}
return x == 0 ? total : -1;
}
}
'코딩테스트 > [백준] 코딩테스트 연습' 카테고리의 다른 글
드래곤 커브 - 15685번 (0) | 2022.04.07 |
---|---|
미네랄 - 2933번 (0) | 2022.04.06 |
스도쿠 - 2239번 (0) | 2022.04.06 |
사다리 조작 - 15684번 (0) | 2022.04.05 |
톱니바퀴 - 14891번 (0) | 2022.04.04 |