poj 2236 并查集（修电脑，使两两互联）

最新推荐文章于 2020-05-19 22:37:50 发布

dumeichen

最新推荐文章于 2020-05-19 22:37:50 发布

阅读量378

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/dumeichen/article/details/43604205

数据结构专栏收录该内容

50 篇文章

订阅专栏

本文讨论了如何使用并查集解决电脑维修问题，包括电脑的位置、维修操作及测试连接是否正常。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给定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;
}