POJ 2236

#include<stdio.h>
int pa[1010],vis[1010];
struct p
{
    int x,y;
}num[1011];
int n,d;
int find(int k)//并查集的查找
{
    return pa[k]==k?k:pa[k]=find(pa[k]);
}
int bing(int i,int p)//并查集的合并
{
    p=find(p);
    i=find(i);
    if(i!=p)
        pa[p]=i;
}
int sum(int i,int j)//计算距离是否达到
{
    int x1=num[i].x-num[j].x;
    int x2=num[i].y-num[j].y;
    if(x1*x1+x2*x2<=d*d)
        return 1;
    return 0;
}
int main()
{
    scanf("%d%d",&n,&d);
    for(int i=1;i<=n;i++)
    {
        pa[i]=i;
        vis[i]=0;
    }//设置为原始数据
    for(int i=1;i<=n;i++)
        scanf("%d%d",&num[i].x,&num[i].y);
    char ch;
    while(scanf("%c",&ch)==1)
    {
        int x,y;
        if(ch=='O')
        {
            scanf("%d",&x);
            vis[x]=1;
            for(int i=1;i<=n;i++)
            {
                if(i!=x&&sum(i,x)&&vis[i])
                {
                    bing(i,x);
                }
            }
        }
        if(ch=='S')
        {
            scanf("%d%d",&x,&y);
            x=find(x);
            y=find(y);
            if(x==y)
                puts("SUCCESS");
            else
                puts("FAIL");
        }
    }
    return 0;
}

你可能感兴趣的:(POJ 2236)