BOJ)2610 회의준비

문제: icpc.me/2610

각각의 컴포넌트에서 해당 컴포넌트에 속한 모든 정점으로 가는 최단거리들의 최댓값이 최소가 되는 정점들을 출력하는 문제이다.

각 정점들간의 최단거리는 플로이드 워셜을 이용하여 구해주면 되고 어떤 컴포넌트에 속한 정점에서 해당 컴포넌트에 속한 모든 정점으로 가는 최단거리들 중 최댓값을 힙에 저장하여 순서대로 답에 넣어주는 것으로 문제를 해결할 수 있다.

힙을 볼 때 이미 처리한 컴포넌트에 속한 정점은 지나쳐주어야하는데 이는 disjoint-set을 이용하면 간단하게 해결해줄 수 있다.

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
#define INF 987654321
using namespace std;
int n, m, a[101][101], x, y, p[101], c[101];
vector<int> r;
priority_queue<pair<int, int>> pq;
int f(int h) {
    return h == p[h] ? h : p[h] = f(p[h]);
}
void merge(int x, int y) {
    x = f(x);
    y = f(y);
    if (x == y)return;
    p[x] = y;
}
int main() {
    scanf("%d%d", &n, &m);
    memset(a, 0x3f, sizeof(a));
    for (int i = 1; i <= n; i++)
        p[i] = i;
    for (int i = 0; i < m; i++) {
        scanf("%d%d", &x, &y);
        a[x][y] = 1;
        a[y][x] = 1;
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            for (int k = 1; k <= n; k++) {
                if (a[j][i] + a[i][k] < a[j][k])
                    a[j][k] = a[j][i] + a[i][k];
            }
        }
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++)
            if (a[i][j] != a[0][0])
                merge(i, j);
    }
    for (int i = 1; i <= n; i++) {
        int hc = 0;
        for (int j = 1; j <= n; j++) {
            if (i == j)continue;
            if (a[i][j] == a[0][0])continue;
            hc = max(hc, a[i][j]);
        }
        pq.push({ -hc,i });
    }
    while (pq.size()) {
        int h = pq.top().second;
        pq.pop();
        if (c[f(h)])continue;
        r.push_back(h);
        c[f(h)] = 1;
    }
    sort(r.begin(), r.end());
    printf("%d\n", r.size());
    for (auto x : r)
        printf("%d\n", x);
    return 0;
}