POJ-2236 Wireless Network---并查集

题目链接：

 https://vjudge.net/problem/POJ-2236

题目大意：

给你N台电脑，从1-N。一个数字，表示两台计算机的最大通信距离，超过这个距离就无法进行通信。然后分别告诉这些电脑的坐标，接下来有两种操作，第一种O表示这点电脑修好，第二种S，表示测试这两台电脑能不能进行正常的通信

思路：

首先需要注意，只有某台电脑修好了，这台电脑才可以参加通信。

并查集的简单应用，对每次修好的电脑对其它已经修好的电脑遍历，如果距离小于等于最大通信距离就将他们合并。之后判断2台电脑是不是一个集合中就KO了

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<set>
#include<map>
#include<cmath>
using namespace std;
typedef pair<int, int> Pair;
typedef long long ll;
const int INF = 0x3f3f3f3f;
int T, n, m, d;
const int maxn = 1e3 + 10;
bool vis[maxn];//vis[i] = 1表示第i台电脑已经修好
int pa[maxn];
vector<int>G[maxn];
struct node
{
    int x, y;
}a[maxn];
int dis(node a, node b)
{
    return (a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y);
}
void init()
{
    for(int i = 1; i <= n; i++)scanf("%d%d", &a[i].x, &a[i].y);
    for(int i = 1; i <= n; i++)
    {
        for(int j = i + 1; j <= n; j++)
            if(dis(a[i], a[j]) <= d * d)G[i].push_back(j), G[j].push_back(i);
    }
    for(int i = 0; i <= n; i++)pa[i] = i;
}
int Find(int x)
{
    return x == pa[x] ? x : pa[x] = Find(pa[x]);
}
int main()
{
    cin >> n >> d;
    init();
    string s;
    int x, y;
    while(cin >> s)
    {
        if(s[0] == 'O')
        {
            scanf("%d", &x);
            vis[x] = 1;
            for(int i = 0; i < G[x].size(); i++)
            {
                y = G[x][i];
                if(vis[y])
                {
                    pa[Find(y)] = Find(x);
                    //这里必须是pa[Find(y)] = Find(x)不能是pa[y] = x,因为是并查集根节点的更改
                }
            }
        }
        else if(s[0] == 'S')
        {
            scanf("%d%d", &x, &y);
            //cout<<Find(x)<<" "<<Find(y)<<endl;
            if(Find(x) == Find(y))
            {
                printf("SUCCESS
");
            }
            else printf("FAIL
");
        }
    }
}