POJ-2236 Wireless Network（并查集深度优化模板题）

点击查看原题
题目大意：有n台电脑，电脑最大联机距离为d，给出每个电脑的坐标，进行多个操作，O p操作修复p号电脑，S p，q操作 询问p号电脑和q号电脑是否可以连接。
分析：
因为时间给了10s所以是并查集模板题。
这里使用了深度优化让rank小的拼接到rank大的上面。

#include<stdio.h>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;
int n,d;
struct point
{
    int x, y;
    point(){}
    point(int xx,int yy)
    {
        x=xx,y=yy;
    }
}edge[100021];
bool repair[100021];
int father[100021];
int rank1[100021];
void init()
{
    for(int i=0;i<=n;i++)
    {
        father[i]=i;
        rank1[i]=0;
    }
}
int find(int a)
{
    return a==father[a]?a:find(father[a]);
}

bool unite(int a,int b)
{
    int aa=find(a);
    int bb=find(b);
    if(aa==bb) return false;
    if(rank1[aa]<rank1[bb])
    {
        father[aa]=bb;
    }
    else
    {
        father[bb]=aa;
        if(rank1[aa]=rank1[bb])
        rank1[aa]++;
    }
    return true;
}
void vjudge(int a,int b)
{
    int aa=find(a);
    int bb=find(b);
    if(aa==bb)
    return;
    else
        if((edge[a].x-edge[b].x)*(edge[a].x-edge[b].x)+(edge[a].y-edge[b].y)*(edge[a].y-edge[b].y)<=d*d)
            unite(a,b);
    return;
}

int main()
{
    while(scanf("%d%d",&n,&d)==2)
    {
        init();
        memset(repair,false,sizeof(repair)); 
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&edge[i].x,&edge[i].y);
        }
        char s[20];
        while(scanf("%s",s)==1)
        {
            if(s[0]=='O')
            {
                int a;
                scanf("%d",&a);
                repair[a]=true;
                for(int i=1;i<=n;i++)
                {
                    if(repair[i])
                    vjudge(a,i);
                }
            }
            else
            {
                int a,b;
                scanf("%d%d",&a,&b);
                if(find(a)==find(b))
                printf("SUCCESS\n");
                else
                printf("FAIL\n");
            }
        }
    } 
}