Poj-2236 Wireless Network

[题目链接]

题意： 现在有n台损坏的电脑，目前可以一个一个地进行维修。由于硬件的限制，修好了的电脑想要直接相连，那么它们之间的距离必须≤dist。但是如果A与C连接，C与B又连接，那么A与B就间接连接。在修电脑的过程中，工作者随时可以进行两种操作，修电脑或者是测试两台电脑是否已经连接。你的任务是对于每一次测试，如果成功连接，输出SUCCESS，否则输出FAIL。

思路： 还不错的并查集题，先预处理两两电脑是否可达，然后每当修复一台电脑时，遍历n台电脑，合并和它可达且修复的电脑即可。

时间限制10s真是很出乎意料啊~~

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int Max_n=1100;

int n,d;
int par[Max_n];   
int rank[Max_n];
int pos[Max_n][2]; //电脑坐标
bool used[Max_n];  //电脑是否修复
bool vis[Max_n][Max_n]; //i电脑到j电脑是否可达

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

void unite(int x,int y){
    x=find(x);y=find(y);
    if(x==y)return;
    if(rank[x]<rank[y])par[x]=y;
    else par[y]=x;
    if(rank[x]==rank[y])rank[x]++;
}

int main()
{
    scanf("%d%d",&n,&d);
    for(int i=0;i<=n;i++){
        par[i]=i;
        rank[i]=0;
    }
    memset(vis,0,sizeof(vis));
    memset(used,0,sizeof(used));
    for(int i=1;i<=n;i++)
        scanf("%d%d",&pos[i][0],&pos[i][1]);
    for(int i=1;i<=n;i++){ //预处理所有两两电脑是否可达
        for(int j=i+1;j<=n;j++){
            double f=sqrt((double)(pos[i][0]-pos[j][0])*(pos[i][0]-pos[j][0])+(pos[i][1]-pos[j][1])*(pos[i][1]-pos[j][1]));
            if(f<=d){
                vis[i][j]=1;vis[j][i]=1;
            }
        }
    }
    char s[3];
    int a,b;
    while(~scanf("%s",s)){
        if(s[0]=='O'){
            scanf("%d",&a);
            for(int i=1;i<=n;i++) //i电脑已经修复且a电脑到i电脑可达，则合并
                if(used[i]&&vis[a][i])unite(a,i);
            used[a]=1;
        }
        else {
            scanf("%d%d",&a,&b);
            a=find(a);b=find(b);
            if(a==b)printf("SUCCESS\n");
            else printf("FAIL\n");
        }
        getchar();
    }
    return 0;
}