POJ ~ 2236 ~ Wireless Network （并查集）

题意：地震了，电脑都坏了不工作了，现在要修复这些电脑，和询问这些电脑是否可以通信（如果距离不大于D就可以通信，超过D则不行）先输入N,D；N表示有多少台电脑，D表示距离；然后N行输入这些电脑所在的X，Y坐标；下面有一些询问或者修复操作以EOF结尾（在这里错了好几次）；

《1》如果第一个字母为‘S’则后边跟两个数字p，q表示询问p和q之间能不能通信，能通信输出SUCCESS，不能输出FAIL；

《2》如果第一个字母为‘O’则后边跟一个数字p，表示修复p这台电脑；（一开是没看懂题，因为把这个O跟0（零）看成一样的）

思路：并查集；用结构体存这些电脑信息，X,Y坐标以及他的父亲节点pre和flag表示有没有修复好；输入的时候初让每个电脑的父亲节点等于自己；flag = false，表示都没修好。

如果为S的话直接判断p，q的父亲节点是否一样即可；如果为O的话，让compter[p]等于1表示修复，把这个电脑跟所有的电脑进行合并操作，能合并的条件为两台电脑都被修复且距离不超过d。注意操作的输入是以EOF结尾的。

//#include<bits/stdc++.h>
#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN = 1005;
int n, d;
struct Compter
{
    int x, y, pre;
    bool flag;
}compter[MAXN];
int find(int x)
{
    if (compter[x].pre == x) return x;
    compter[x].pre = find(compter[x].pre);
    return compter[x].pre;
}
void Union(int a, int b)
{
    int root1 = find(a), root2 = find(b);
    if (compter[a].flag && compter[b].flag &&
        (compter[a].x - compter[b].x) * (compter[a].x - compter[b].x) +
        (compter[a].y - compter[b].y) * (compter[a].y - compter[b].y) <= d * d )
    {
        compter[root1].pre = root2;
    }

}
int main()
{
    while (~scanf("%d%d", &n, &d))
    {
        for (int i = 1; i <= n; i++)
        {
            scanf("%d%d", &compter[i].x, &compter[i].y);
            compter[i].pre = i;
            compter[i].flag = false;
        }
        char op;
        int p, q;
        while (~scanf("%c", &op))
        {
            if (op == 'O')
            {
                scanf("%d", &p);
                compter[p].flag = true;
                for (int i = 1; i <= n; i++) Union(i, p);
            }
            else if(op == 'S')
            {
                scanf("%d%d", &p, &q);
                if (find(p) == find(q)) printf("SUCCESS\n");
                else printf("FAIL\n");
            }
        }
    }
    return 0;
}
/*
4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4
*/