poj2236 Wireless Network

每修好一台电脑后就逐一与之前修好的电脑进行距离计算，若距离小于等于d则加入同一集合。

写的时候各种不认真，wa了好几次，应该好好批评下自己。

#include<cstdio>
#include<string.h>
#include<iostream>
#include<math.h>
const int maxn=1010;
int p[maxn],n,d,repair[maxn],k;
struct Node
{
    double x,y;
    bool flag;
}node[maxn];

void init()
{
    int i;
    for(i=1;i<=n;i++)
    {
        p[i]=i;
        node[i].flag=false;
    }
    k=0;
}

double dis(int a,int b)
{
    double L=(node[a].x-node[b].x)*(node[a].x-node[b].x)*(1.0);
    L+=(node[a].y-node[b].y)*(node[a].y-node[b].y)*(1.0);
    L=sqrt(L);
    return L;
}

int find(int x)
{
    if(x!=p[x])
    {
        p[x]=find(p[x]);
    }
    return p[x];
}

void unions(int a,int b)
{
    int ra=find(a);
    int rb=find(b);
    if(ra==rb)
    return;
    p[ra]=rb;
}
int main()
{
    //freopen("test.txt","r",stdin);
    scanf("%d%d",&n,&d);
    int i,a,b,ra,rb;
    for(i=1;i<=n;i++)
    {
        scanf("%lf%lf",&node[i].x,&node[i].y);
    }
    init();
    char c[2];
    while(scanf("%s",c)!=EOF)
    {
        if(c[0]=='O')
        {
            scanf("%d",&repair[k]);
            node[repair[k]].flag=true;
            for(i=0;i<k;i++)
            {
                if(dis(repair[i],repair[k])<=d)
                unions(repair[i],repair[k]);
            }
            k++;
        }
        else
        {
            scanf("%d%d",&a,&b);
            ra=find(a);
            rb=find(b);
            if(ra==rb)
            {
                printf("SUCCESS\n");
            }
            else
            printf("FAIL\n");
        }
    }
    return 0;
}