POJ 2236 Wireless Network

并查集

题意

若两台修好的计算机之间的距离小于等于d时可以直接通信；
若两个计算机A、B和中间计算机C满足（A和C的距离，B和C的距离小于等于d），则A和B可以通信
O i 表示将i号计算机修好； S i j 问i和j是否能通信

题解

使用并查集。

“O i”： i没有所属集合时（par[i] == i）会将i加入和它相连且已维修节点的所属集合，后续会继续合并i所属集合 和 i相连集合

“S i j” ：查看i和j是否属于一个集合

代码

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
const int MAXN = 10010;

int N,d;
struct Point{
    int x;
    int y;
    int par;
};
Point comp[MAXN];
bool repair[MAXN];

int find_root(int x)
{
    int r = x;
    while(comp[r].par != r)
        r = comp[r].par;
    int i = x;
    while(i != r) // compress the path
    {
        int t = comp[i].par;
        comp[i].par = r;
        i = t;
    }
    return r;
}
void join(int node)
{
    repair[node] = true;
    for(int i = 1; i <= N; i++)
    {
        if(i != node && repair[i])
        {
            double dis = pow(comp[i].x - comp[node].x, 2) + pow(comp[i].y - comp[node].y, 2) ;
            if(dis <= d * d)
            {
                int root1 = find_root(i);
                int root2 = find_root(node);
                if(root1 != root2)
                {
                    comp[root2].par = root1;
                }
            }


        }
    }
}

int main()
{
    cin >> N >> d;
    for(int i = 1; i <= N; i++)
    {
        cin >> comp[i].x >> comp[i].y;
        comp[i].par = i;
    }
    memset(repair, false, sizeof(repair));
    char temp[MAXN];
    while(cin >> temp)
    {
        switch(temp[0])
        {
            case 'O':
                int opr;
                cin >> opr;
                join(opr);
                break;
            case 'S':
                int x,y;
                cin >> x >> y;
                if(find_root(x) == find_root(y))
                {
                    cout << "SUCCESS" <<endl;
                }
                else
                    cout << "FAIL" <<endl;
                break;

        }
    }
    return 0;
}