취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 5081번 문제인 Constellations이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/5081

$n$이 500 이하로 충분히 작은 점을 눈여겨보자. 이 정도의 제한이면 각각의 별에 대하여 그 별과 가장 가까운 별(들)을 다른 별들을 한번씩 확인하는 것으로 충분히 구할 수 있다.

각 별에 대하여 가장 가까운 별들의 모임을 구하고, "한 별이 다른 별의 가장 가까운 별 중 하나"의 관계를 만족하는 두 별을 하나의 별자리로 합쳐 문제를 해결하자. 이 과정은 disjoint set 자료구조를 이용하면 쉽게 구현 가능하다.

아래는 제출한 소스코드이다.

#include <iostream>
#include <vector>
using namespace std;

int N;
int X[500], Y[500];
int uf[500];
int findf(int x) {
    if (x == uf[x]) return x;
    return uf[x] = findf(uf[x]);
}
void solve() {
    for (int i = 0; i < N; i++) uf[i] = i;
    for (int i = 0; i < N; i++) {
        cin >> X[i] >> Y[i];
    }
    for (int i = 0; i < N; i++) {
        int mn = 1000000007; vector<int> vec;
        for (int j = 0; j < N; j++) {
            if (i == j) continue;
            int dx = X[i] - X[j], dy = Y[i] - Y[j];
            int val = dx * dx + dy * dy;
            if (val < mn) {
                vec.clear();
                mn = val, vec.emplace_back(j); 
            }
            else if (val == mn) vec.emplace_back(j);
        }
        for (auto &x:vec) {
            int y = findf(i), z = findf(x);
            uf[z] = y;
        }
    }
    int ans = 0;
    for (int i = 0; i < N; i++) {
        if (i == findf(i)) ans++;
    }

    cout << ans << " constellations.\n";
}

int T;

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> N;
    while (N) {
        T++;
        cout << "Sky " << T << " contains ";
        solve();
        cin >> N;
    }
}