HDU2236 Wireless Network【并查集】

题目链接POJ-2236

kuangbin带你飞【专题五 并查集】

题目描述

有一个计算机网络的所有线路都坏了,网络中有n台计算机,现在你可以做两种操作,修理(O)和检测两台计算机是否连通(S),只有修理好的计算机才能连通。连通有个规则,两台计算机的距离不能超过给定的最大距离D(一开始会给你n台计算机的坐标)。检测的时候输出两台计算机是否能连通。

思路:

 每次修理好一台计算机的时候就遍历一下所有修好的计算机,看距离是否<=D,如果符合说明可以连通,将两台计算机所在集合合并。每次检查的时候判断一下这两台计算机是否在同一集合中即可。

易错点

1.输出是“FAIL” 不是 “FALL”
2.先判断距离是否满足,然后进行Union操作; 如果先Find根, 然后判断距离会wa掉(这样子就是错误的)。
比如 :

1
2
3
4
5
6
7
8
9
10
void Union(int x, int y)
{
x = Find(x);
y = Find(y);
if(x != y)
{
if(pow(p[x].x - p[y].x, 2) + pow(p[x].y - p[y].y, 2) <= d * d)
p[x].pre = y;
}
}

这样就会wa掉,原因很简单分析出来!

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#include <iostream>
#include <cstdlib>
#include <cmath>
#include <cstring>
using namespace std;

#define MAX_N 1010

bool vis[MAX_N];//标记vis[k], 表示标号为k电脑的电脑是否可用
int n, d;

struct Node{
int x, y;
int pre;//pre 代表祖宗
}p[MAX_N];

void init(int n)
{
for(int i = 1; i<=n; i++)
p[i].pre = i;
}

int Find(int x)
{
return x == p[x].pre ? x : p[x].pre = Find(p[x].pre);//路径压缩
}

void Union(int x, int y)
{
x = Find(x);
y = Find(y);
if(x != y)
{
p[x].pre = y;
}
}

int main()
{
cin >> n >> d;
init(n);
for(int i = 1; i<=n; i++)
{
cin >> p[i].x >> p[i].y;
}
memset(vis, 0, sizeof(vis));
string s;
while(cin >> s)
{
if(s == "O")
{
int k;
cin >> k;
vis[k] = 1;//标记可用
for(int i = 1; i<=n; i++)//如果之前有可用 而且距离合适 则进行合并
{
if(vis[i] && i != k && pow(p[i].x - p[k].x, 2) + pow(p[i].y - p[k].y, 2) <= d * d)
{
Union(i, k);
}
}
}
else
{
int x, y;
cin >> x >> y;
if(Find(x) == Find(y))//判断是否是同一个集合
{
cout<<"SUCCESS"<<endl;
}
else
{
cout<<"FAIL"<<endl;
}
}
}
return 0;
}
-------------本文结束,感谢您的阅读!-------------