팀 조합을 생성하고, 차를 구한 다음 값의 크기를 비교한다.
이를 코드로 어떻게 구현할까?
1. 팀 조합 생성하기
private static void combi(int i, int depth) {
if (depth == N / 2) {
diff();
return;
}
for (int j = i; j < N; j++) {
visit[j] = true;
combi(j + 1, depth + 1);
visit[j] = false;
}
}
visit[]
배열은 해당 팀원 방문 여부를 나타낸다.depth
는 깊이를 몇명까지 확인했는지를 의미한다.depth
가 팀원 수의 절반이라면 조합이 생성되었다는 의미이니, 생성된 조합으로 value를 구한다.2. 생성된 조합으로 value 구하기
private static void diff() {
int start_team = 0;
int link_team = 0;
for (int i = 0; i < N; i++) {
for (int j = i; j < N; j++) {
if (visit[i] == true && visit[j] == true) {
start_team += map[i][j];
start_team += map[j][i];
} else if (visit[i] == false && visit[j] == false) {
link_team += map[i][j];
link_team += map[j][i];
}
}
}
value = Math.abs(start_team - link_team);
min = Math.min(value, min);
}
start_team
과 link_team
은 각각의 크기를 말한다.visit[]
배열에서 true
, false
를 기준으로 조합이 생성되었다. 이를 확인하여 start_team
과 link_team
의 크기를 각각 구한다.value
를 찾고 작은 값을 min
으로 지정한다.package Algorithm.P14889;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class P14889 {
static int N;
static int[][] map;
static boolean[] visit;
static int min = Integer.MAX_VALUE;
static int value;
static StringBuilder sb;
public static void main(String[] args) throws IOException {
System.setIn(new FileInputStream("src/Algorithm/P14889/input.txt"));
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
sb = new StringBuilder();
N = Integer.parseInt(st.nextToken());
map = new int[N][N];
visit = new boolean[N];
for (int i = 0; i < N; i++) {
st = new StringTokenizer(br.readLine());
for (int j = 0; j < N; j++) {
map[i][j] = Integer.parseInt(st.nextToken());
}
}
combi(0, 0);
sb.append(min);
System.out.println(sb);
}
private static void combi(int i, int depth) {
if (depth == N / 2) {
diff();
return;
}
for (int j = i; j < N; j++) {
visit[j] = true;
combi(j + 1, depth + 1);
visit[j] = false;
}
}
private static void diff() {
int start_team = 0;
int link_team = 0;
for (int i = 0; i < N; i++) {
for (int j = i; j < N; j++) {
if (visit[i] == true && visit[j] == true) {
start_team += map[i][j];
start_team += map[j][i];
} else if (visit[i] == false && visit[j] == false) {
link_team += map[i][j];
link_team += map[j][i];
}
}
}
value = Math.abs(start_team - link_team);
min = Math.min(value, min);
}
}