并查集＿无线网络

题目描述
一场地震在东南亚发生了。不幸的是ACM组织通过电脑建立的无线网络遭到了毁灭性的影响—--网络中所有的电脑都损坏了。在陆续的维修电脑之后，无线网络有逐渐开始再一次运作了。由于硬件的制约，每两台电脑只能保持不超过d米的距离，才可以直接进行通讯。但是每台电脑又可以作为其它两台电脑通讯的中介点，也就是说假设A电脑与B电脑不在能直接通讯的的范围内，但是它们可以通过同时能与A和B电脑通讯的C电脑建立间接通讯关系。
在维修的过程中，维修者可以进行两种操作：维修一台电脑或者检测两台电脑之间是否能够通讯。你的任务就是解答每一次的检测操作。
输入
第一行包含两个整数N和d (1 <= N <= 1001, 0 <= d <= 20000).N表示电脑的数量，电脑编号从1开始到N，d为两台能直接通讯的电脑所需保持的距离的最大值。在接下来的N行里，每行包含两个整数xi, yi (0 <= xi, yi <= 10000),表示N台电脑的坐标。接下来的一系列的输入都表示维修者的操作，每种操作都是以下两种中的一种：
1. "O p" (1 <= p <= N),表示维修第p台电脑。
2. "S p q" (1 <= p, q <= N), 表示检测p与q电脑是否能够通讯。
输入不会超过300000 行。 
输出
对于每组检测操作，若两台电脑能进行通讯就输出"SUCCESS"，否则输出"FAIL"。
样例输入
4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4
样例输出
FAIL

SUCCESS

#include <iostream>#include <cstdio>#include <cmath>#include <cstring>using namespace std;#define N 1005int flag[N],imap[N][N];   //flag 判断是否维修  imap判断是否可以直接联通，不借助任何中间点int parent[N],x[N],y[N];  //存储根节点与位置int mul(int x){    return x*x;}int root(int x){     return   parent[x] == x ?  x : parent[x]=root( parent[x] );}void make(int a,int b)//并查集的并操作{    int f1=root(a);    int f2=root(b);    if(f1!=f2)        parent[f2]=f1;}void check(int a,int b){    if(root(a)==root(b))        printf("SUCCESS\n");    else        printf("FAIL\n");}int main(){    int n,m;    double d;    scanf("%d%lf",&n,&d);    memset(flag,0,sizeof(flag));    memset(imap,0,sizeof(imap));    for(int i=1;i<=n;i++){      scanf("%d%d",&x[i],&y[i]);      parent[i]=i;    }    for(int i=1;i<=n;i++)      for(int j=i+1;j<=n;j++)         if(sqrt(mul(x[i]-x[j])+mul(y[i]-y[j]))<=d){            imap[i][j]=1;            imap[j][i]=1;  //从i到j或者从j到i可以直接联通         }     char s[10];     while(scanf("%s",s)!=EOF)     {    int a,b;         if(strcmp(s,"O")==0)         {             scanf("%d",&a);             flag[a]=1; //位置为ipos的节点进行了维修             for(int i=1;i<=n;i++)if(imap[i][a]&&flag[i])  //将所有已经维修好的站点与a进行比较判断是否可以联通make(a,i);         }         else         {             scanf("%d%d",&a,&b);             check(a,b);         }     }     return 0;}