POJ-2236 Wireless Network (简单并查集)

也是几个月前做的题目,填坑

POJ-2236 Wireless Network

Description

南亚发生了一次地震。ACM (Asia Cooperated Medical 亚洲联合医疗队) 已经为膝上型电脑搭建了一个无线网络,但受到了一次不可预知的余震攻击,因此网络中的所有电脑都被破坏了。电脑被逐台修复,网络逐步恢复了工作。由于受到硬件的约束,每台电脑只能与距离它不超过 d 米的其它电脑直接通信。但每台电脑可被看作其它两台电脑的通信中转点,也就是说,如果电脑 A 和电脑 B 可以直接通信,或存在一台电脑 C 既可与 A 也可与 B 通信,那么电脑 A 和电脑 B 之间就能够通信。

在处理网络修复的过程中,工作人员们在任何一个时刻,可以执行两种操作:维修一台电脑,或测试两台电脑是否能够通信。请您找出全部的测试操作。

Input

第一行包含了两个整数 N 和 d (1 <= N <= 1001, 0 <= d <= 20000)。此处 N 是电脑的数目,编号从 1 到 N;同时,D 是两台电脑之间能够直接通信的最大距离。接下来的 N 行,每行包含两个整数 xi, yi (0 <= xi, yi <= 10000),表示 N 台电脑的坐标。从第 (N+1) 行到输入结束,是逐一执行的操作,每行包含一个操作,格式是以下两者之一:

  1. “O p” (1 <= p <= N),表示维护电脑 p 。
  2. “S p q” (1 <= p, q <= N),表示测试电脑 p 和 q 是否能够通信。

输入不超过 300000 行。

Output

对于每个测试操作,如果两台电脑能够通信,则打印 “SUCCESS”;否则,打印 “FAIL”。

Sample Input

4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4

Sample Output

FAIL
SUCCESS

AC代码:

#include<bits/stdc++.h>
#define maxn 1010
using namespace std;

struct com{
   int x,y;
   bool isok;
};

struct com a[maxn];
int pre[maxn];
double dist[maxn][maxn];


int find_r(int x)
{
    if(pre[x]==x) return x;
    else{
        int root = find_r(pre[x]);
        pre[x] = root;
        return root;
    }//寻根同时化简并查集
}

void unite(int x, int y)
{
    x = find_r(x); y = find_r(y);
    pre[x] = y;
    return;
}//合并两个并查集

int N,d;

int main()
{
    scanf("%d%d",&N,&d);
    for(int i = 1; i <= N; i++)
    {
        scanf("%d %d",&a[i].x, &a[i].y);
        pre[i] = i; a[i].isok = false;
    }
    
    // 计算各电脑之间的距离
    for(int i = 1; i <= N; i++)
        for(int j = i; j <= N; j++)
    {
        if(i!=j)
            dist [j][i] = dist[i][j] = sqrt((a[i].x - a[j].x)*(a[i].x - a[j].x) + (a[i].y - a[j].y)*(a[i].y - a[j].y));
        else dist[i][j] = 0;
    }
    
    char op;
    getchar();
    while(scanf("%c",&op)!=EOF)
    {
        if(op=='O'){
            int k;
            scanf("%d",&k);
            a[k].isok = true;
            for(int i = 0; i <= N; i++)
                if(dist[k][i] <= d && a[i].isok && i!=k) // 一定要两台电脑都正常工作才能合并
                    unite(k,i);
        }
        else {
            int x,y;
            scanf("%d %d",&x, &y);
            x = find_r(x); y = find_r(y);
            if(x==y && a[x].isok && a[y].isok) puts("SUCCESS"); // 一定要两台电脑都正常工作才能合并
            else puts("FAIL");
        }
        getchar();
    }

    return 0;
}

发布了35 篇原创文章 · 获赞 15 · 访问量 1121

猜你喜欢

转载自blog.csdn.net/irimsky/article/details/96841099