题目链接
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 <cstdio>
#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
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;i<g[u].size();i++)
{
int 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;
}