Wireless Network（并查集（裸））

题目来源：https://vjudge.net/problem/POJ-2236
【题意】
因为地震，所有电脑都损坏，给出n个电脑的位置坐标，然后说明只有在有限距离内两台电脑才能相互连接，一系列操作之后，多次询问。
操作包括两种： 1.成功修复一台电脑。O +序号。
2.询问x和y电脑是否相连。 S x y。
【思路】
给出一个修复好的电脑，就把它和所有可以连接的电脑连在一起。
【代码】

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<map>
#include<queue>
#include<stack>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int n;
double D;
int pre[1010];
struct pp
{
    double x,y;
} s[1010];
bool dis(double x1,double y1,double x2,double y2)
{
    double p=sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
    return p<=D?1:0;
}
int find(int root)
{
    if(root==pre[root])
        return root;
    else return pre[root]=find(pre[root]);
}
void MIX(int a,int b)
{
    int fa=find(a);
    int fb=find(b);
    if(fa!=fb)
        pre[fa]=fb;
}
int main()
{
    scanf("%d%lf",&n,&D);
    mem(pre,0);
    for(int i=1; i<=n; i++)
        scanf("%lf%lf%*c",&s[i].x,&s[i].y);
        char p;
    while(~scanf("%c",&p))
    {
        if(p=='O')
        {
            int m;
            scanf("%d%*c",&m);
            pre[m]=m;
            for(int i=1; i<=n; i++)
                if(pre[i]&&dis(s[i].x,s[i].y,s[m].x,s[m].y))
                        MIX(i,m);
        }
        else if(p=='S')
        {
            int m,ml;
            scanf("%d%d%*c",&m,&ml);
            if(find(m)==find(ml))
                printf("SUCCESS\n");
            else
                printf("FAIL\n");
        }
    }
}