POJ2236詳細的解題報告（C語言版）

#include<stdio.h>
#include<string.h>
int a[100000];
int flag[100000];//標記電腦是否被修，1為修好，0為未修
int b[100000][2];//座標
int n;
void creat()
{
    for(int i=1;i<=n;i++)//由於題目要求N>0，編號大於0
    {
        a[i]=i;
    }
}
int set_friend(int x)//找祖先
{
    int r=x;
    while(a[r]!=r)
    {
        r=a[r];
    }
    int i=x;
    int j;
    while(i!=r)//路徑壓縮

    {
        j=a[i];
        a[i]=r;
        i=j;
    }
    return r;
}
void join_friend(int x,int y)//連起來
{
    int fx=set_friend(x);
    int fy=set_friend(y);
    if(fx!=fy)
    {
        a[fx]=fy;
    }
}
int is_friend(int x,int y)//判斷電腦是否可以連通
{

    int fx=set_friend(x);
    int fy=set_friend(y);
    if(fx==fy)
    return 1;
    else return 0;
}
int distance(int x,int y)//求兩臺電腦之間距離的平方
{
    return ((b[x][0]-b[y][0])* (b[x][0]-b[y][0])+(b[x][1]-b[y][1])*(b[x][1]-b[y][1]));
}
int main()
{
        int d;
        char  c[10];
        while( scanf("%d%d",&n,&d)!=EOF)
        {
                creat();
                memset(flag,0,sizeof(flag));
                for(int i=1;i<=n;i++)
                {
                scanf("%d%d",&b[i][0],&b[i][1]);
                }
                d*=d;//距離的平方進行判斷，方便
                while(scanf("%s",c)!=EOF)
                {
                    if(c[0]=='O')
                    {
                        int t;
                        scanf("%d",&t);
                        flag[t]=1;
                        for(int j=1;j<=n;j++)
                        {
                            if(flag[j]==1&&distance(j,t)<=d)
                            {
                                join_friend(j,t);
                            }
                        }

                    }
                    if(c[0]=='S')
                    {
                    int a,b;
                    scanf("%d%d",&a,&b);
                    if( is_friend(a,b)==1)
                    printf("SUCCESS\n");
                    else if(is_friend(a,b)==0) printf("FAL\n");
                    }
                }
            }




    return 0;
}