POJ2236 Wireless Network (并查集判断连通性）

题目链接

http://poj.org/problem?id=2236

题目

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

在处理网络修复的过程中，工作人员们在任何一个时刻，可以执行两种操作：维修一台电脑，或测试两台电脑是否能够通信。请您找出全部的测试操作。
输入
第一行包含了两个整数 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 行。
输出
对于每个测试操作，如果两台电脑能够通信，则打印 “SUCCESS”；否则，打印 “FAIL”。
示例输入
4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4
示例输出
FAIL
SUCCESS

分析

先建一个图，表示电脑都是好的时候能相互通信的网络。
对于维修操作，将被维修的电脑和与其相连的且为好的电脑合并到一个并查集中。
对于查询操作，判断两者是否在一个集合中即可。

AC代码

//1204ms 4.6MB
#include 
#include 
#include 
#include 
#include 
using namespace std;

const int maxn=1e3+100;
vector<int> g[maxn];
int N,par[maxn];
int d;
bool vis[maxn];
struct node
{
    int x,y;
}a[maxn];
double disp(node u,node v)
{
    return (u.x-v.x)*(u.x-v.x)+(u.y-v.y)*(u.y-v.y);
}

int find(int x)
{
    if(par[x]==x) return x;
    return par[x]=find(par[x]);
}
bool same(int x,int y)
{
    return find(x)==find(y);
}

void unit(int x,int y)
{
    int fx=find(x),fy=find(y);
    if(fx!=fy)
    {
        par[fx]=fy;
    }
}

void init()//初始化并查集和网络
{
    for(int i=0;i<=N;i++) par[i]=i;
    for(int i=0;i<=N;i++) g[i].clear();
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=N;i++)
        for(int j=i+1;j<=N;j++)
    {
        if(disp(a[i],a[j])<=d*d)
        {
            g[i].push_back(j);
            g[j].push_back(i);
        }
    }
}
void update(int u)
{
    vis[u]=true;
    for(int i=0;iint v=g[u][i];
        if(vis[v]&&!same(u,v))
            unit(u,v);
    }
}
int main()
{
    scanf("%d%d",&N,&d);
    for(int i=1;i<=N;i++)
        scanf("%d%d",&a[i].x,&a[i].y);
    char op[2];
    init();
    while(~scanf("%s",op))
    {
        int p,q;
        if(op[0]=='O')
        {
            scanf("%d",&p);
            update(p);
        }
        else
        {
            scanf("%d%d",&p,&q);
            printf("%s\n",same(p,q)?"SUCCESS":"FAIL");
        }
    }
    return 0;
}