POJ2236(并查集)

题目：http://poj.org/problem?id=2236

 

题意：给定n个点的坐标，然后选出其中的一些点出来，问在这些点中的指定的两点是否连通。

#include <iostream>
#include <string.h>
#include <stdio.h>

using namespace std;
const int N = 1005;

struct Point
{
    int x,y;
};

Point p[N];
int repaired[N];
int pre[N],rank[N];

int dist(Point A,Point B)
{
    return (A.x-B.x)*(A.x-B.x)+(A.y-B.y)*(A.y-B.y);
}

void Init(int n)
{
    for(int i=1;i<=n;i++)
    {
        pre[i] = i;
        rank[i] = 1;
    }
}

int Find(int x)
{
    if(pre[x] != x)
        pre[x] = Find(pre[x]);
    return pre[x];
}

void Union(int x,int y)
{
    x = Find(x);
    y = Find(y);
    if(x == y) return;
    if(rank[x] >= rank[y])
    {
        pre[y] = x;
        rank[x] += rank[y];
    }
    else
    {
        pre[x] = y;
        rank[y] += rank[x];
    }
}

int main()
{
    char ch[5];
    int n,d,x,y;
    scanf("%d%d",&n,&d);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&p[i].x,&p[i].y);
    int cnt = 0;
    Init(n);
    while(~scanf("%s",ch))
    {
        if(ch[0] == 'O')
        {
            scanf("%d",&x);
            for(int i=0;i<cnt;i++)
            {
                if(dist(p[repaired[i]],p[x]) <= d*d)
                    Union(repaired[i],x);
            }
            repaired[cnt++] = x;
        }
        else
        {
            scanf("%d%d",&x,&y);
            x = Find(x);
            y = Find(y);
            if(x == y) puts("SUCCESS");
            else       puts("FAIL");
        }
    }
    return 0;
}