poj 2236 并查集(修电脑,使两两互联)

题意:给定N台初始都坏了的电脑。一个数字d,表示两台计算机能够直接通信的最大距离,超过这个距离就无法进行通信。然后分别告诉这些电脑的坐标,接下来有两种操作,第一种O表示这台电脑修好,第二种S,表示测试这两台电脑能不能进行正常的通信。

思路:并查集简单应用。每修好一台电脑,将与其能够直接相连的并且已经修好的电脑merge一下。

#include <stdio.h>
#include <string.h>
#include <math.h>
#define N 1005
struct node{
    int x,y;
}p[N];
int n,d;
int f[N],g[N][N],used[N];
double dis(int a,int b){
    return sqrt((p[a].x-p[b].x)*(p[a].x-p[b].x)+(p[a].y-p[b].y)*(p[a].y-p[b].y));
}
int test(int x,int y){
    return dis(x,y) <= d;
}
int find(int x){
    if(f[x] == x)
        return x;
    return f[x] = find(f[x]);
}
void merge(int x,int y){
    f[find(x)] = find(y);
}
int main(){
    int i,j;
    char ch;
    memset(g, 0, sizeof(g));
    memset(used, 0, sizeof(used));
    scanf("%d %d",&n,&d);
    for(i = 1;i<=n;i++)
        f[i] = i;
    for(i = 1;i<=n;i++)
        scanf("%d %d\n",&p[i].x,&p[i].y);
    for(i = 1;i<n;i++)
        for(j = i+1;j<=n;j++)
            if(test(i,j))
                g[i][j] = g[j][i] = 1;
    while((ch=getchar()) != EOF){
        if(ch == 'O'){
            scanf("%d\n",&i);
            used[i] = 1;
            for(j = 1;j<=n;j++)
                if(used[j] && g[i][j])
                    merge(i,j);
        }else{
            scanf("%d %d\n",&i,&j);
            if(used[i] && used[j] && find(i)==find(j))
                printf("SUCCESS\n");
            else
                printf("FAIL\n");
        }
    }
    return 0;
}


你可能感兴趣的:(poj 2236 并查集(修电脑,使两两互联))