[백준] 자바 2610 회의준비

문제

KOI 준비를 위해 회의를 개최하려 한다. 주최측에서는 회의에 참석하는 사람의 수와 참석자들 사이의 관계를 따져 하나 이상의 위원회를 구성하려고 한다. 위원회를 구성하는 방식은 다음과 같다.

1. 서로 알고 있는 사람은 반드시 같은 위원회에 속해야 한다.
2. 효율적인 회의 진행을 위해 위원회의 수는 최대가 되어야 한다.

이런 방식으로 위원회를 구성한 후에 각 위원회의 대표를 한 명씩 뽑아야 한다. 각 위원회의 대표만이 회의 시간 중 발언권을 가지며, 따라서 회의 참석자들이 자신의 의견을 말하기 위해서는 자신이 속한 위원회의 대표에게 자신의 의견을 전달해야 한다. 그런데 각 참석자는 자신이 알고 있는 사람에게만 의견을 전달할 수 있어 대표에게 의견을 전달하기 위해서는 때로 여러 사람을 거쳐야 한다. 대표에게 의견을 전달하는 경로가 여러 개 있을 경우에는 가장 적은 사람을 거치는 경로로 의견을 전달하며 이때 거치는 사람의 수를 참석자의 의사전달시간이라고 한다.

위원회에서 모든 참석자들의 의사전달시간 중 최댓값이 최소가 되도록 대표를 정하는 프로그램을 작성하시오.

예를 들어 1번, 2번, 3번 세 사람으로 구성되어 있는 위원회에서 1번과 2번, 2번과 3번이 서로 알고 있다고 하자. 1번이 대표가 되면 3번이 대표인 1번에게 의견을 전달하기 위해서 2번을 거쳐야만 한다. 반대로 3번이 대표가 되어도 1번이 대표인 3번에게 의견을 전달하려면 2번을 거쳐야만 한다. 하지만 2번이 대표가 되면 1번과 3번 둘 다 아무도 거치지 않고 대표에게 직접 의견을 전달 할 수 있다. 따라서 이와 같은 경우 2번이 대표가 되어야 한다.

입력

첫째 중에 회의에 참석하는 사람의 수 N이 주어진다. 참석자들은 1부터 N까지의 자연수로 표현되며 회의에 참석하는 인원은 100 이하이다. 둘째 줄에는 서로 알고 있는 관계의 수 M이 주어진다. 이어 M개의 각 줄에는 서로 아는 사이인 참석자를 나타내는 두개의 자연수가 주어진다.

출력

첫째 줄에는 구성되는 위원회의 수 K를 출력한다. 다음 K줄에는 각 위원회의 대표 번호를 작은 수부터 차례로 한 줄에 하나씩 출력한다. 한 위원회의 대표가 될 수 있는 사람이 둘 이상일 경우 그중 한 명만 출력하면 된다.

---

테스트 케이스

 

입력 1

9
8
1 2
2 3
3 4
4 5
4 6
4 7
4 8
4 9

출력 1

1

3

 

입력 2

8
7
5 6
4 6
6 7
7 4
1 2
2 3
4 5

출력 2

3

2

4

8

 

입력 3

4
3
2 3
3 4
1 2

출력 3

1

2

 

---

접근

1. 일단 union-find를 이용하여서 각 사람들을 그룹화 해줍니다.

 

2. 플로이드-워셜 알고리즘을 이용해 각 사람간의 거리를 계산 해줍니다.

 

3. 그룹화된 사람들을 개별로 확인하면서 거리의 최대가 가장 짧은 사람을 대표로 선정합니다.

3-1. 문제 예시처럼 1-2-3 이렇게 연결되어 있을때 1번 사람이 대표라면, 2번 사람은 1만큼 3번사람은 2만큼 가야함

3-2. 2번사람이 대표라면 1번 사람과는 1, 3번사람과도 1

3-3. 3번사람이 대표라면 1번 사람과는 2, 3번 사람과는 1

3-4. 따라서 1번사람이 대표가 되면 최대거리는 2, 2번사람이 대표가 되면 최대거리는 1, 3번 사람이 대표가되면 최대거리는 2.... 이것이 바로 문제에서 말하는 최대값의 최소값입니다.

 

4.그룹 개수 출력 + 각 대표를 순서대로 출력합니다.

---

코드

import java.io.*;
import java.util.*;


public class Main {
	static int arr[],N,M;
	public static void makeSet() {
		arr = new int[N+1];
		for(int i=1; i<=N; i++) {
			arr[i] = i;
		}
	}
	public static int findSet(int a) {
		if(a==arr[a]) return a;
		else return arr[a] = findSet(arr[a]);
	}
	public static boolean union(int a, int b) {
		int aRoot = findSet(a);
		int bRoot = findSet(b);
		if(aRoot == bRoot) return false;
		
		arr[bRoot] = aRoot;
		for(int i=1; i<=N; i++) {
			if(arr[i] == bRoot) {
				arr[i] = aRoot;
			}
		}
		return true;
	}
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

		N = Integer.parseInt(br.readLine());
		M = Integer.parseInt(br.readLine());
		boolean visit[] = new boolean[N+1];
		int map[][] = new int[N+1][N+1];
		for(int i=1; i<=N; i++) {
			for(int j=1; j<=N; j++) {
				if(i==j) continue;
				map[i][j] = (int)1e9;
			}
		}
		makeSet();
		int result = 0;
		for(int i=0; i<M; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			int a = Integer.parseInt(st.nextToken());
			int b = Integer.parseInt(st.nextToken());
			if(union(a,b)) result++;
			map[a][b] = 1;
			map[b][a] = 1;
		}
        //플로이드워셜
		for(int k=1; k<=N; k++) {
			for(int i=1; i<=N; i++) {
				for(int j=1; j<=N; j++) {
					if(map[i][j] > map[i][k] + map[k][j]) map[i][j] = map[i][k] + map[k][j];
				}
			}
		}

		//총개수 - union이 성공한 수
		bw.append(String.valueOf(N-result)+"\n");
		int answer[] = new int[N-result];
        //그룹수만큼 반복
		for(int n=0; n<N-result; n++) {
			boolean flag = false;
			ArrayList<Integer> list = new ArrayList<>();
			for(int i=1; i<=N; i++) {
				if(!flag && !visit[i]) {
					arr[0] = arr[i];
					flag = true;
				}
				if(arr[i] == arr[0] && !visit[i]) {
					visit[i] = true;
					list.add(i);
				}
			}
			int min = (int)10e9;
			int max[] = new int[list.size()];
			int saveIndex = 0;
			for(int i=0; i<list.size(); i++) {
				int tempMax = 0;
				for(int j=0; j<list.size(); j++) {
					tempMax = Math.max(tempMax, map[list.get(i)][list.get(j)]);
				}
				max[i] = tempMax;
			}
			for(int i=0; i<max.length; i++) {
				if(max[i] < min) {
					min = max[i];
					saveIndex = list.get(i);
				}
			}
			answer[n] = saveIndex;
		}
        //오름차순 출력
		Arrays.sort(answer);
		for(int i=0; i<N-result; i++) {
			bw.append(String.valueOf(answer[i])+"\n");
		}
		bw.flush();
		bw.close();
		br.close();
	}
}

---

주의

주의할게 많았던 문제...

1. 그룹대표를 오름차순으로 출력할 것.

2. 최대값의 최소값

3. union했을 때 최상단 부모 재설정해주기.