题目链接: vjudge
芜湖! 很开心刚学了并查集就把题做出来了!
题意
南亚发生了一次地震。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) 行到输入结束,是逐一执行的操作,每行包含一个操作,格式是以下两者之一:
- “O p” (1 <= p <= N),表示维护电脑 p 。
- “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代码
#include <stdio.h>
#include <vector>
#include <cstring>
#include <math.h>
#define Num 300005
#define Max 0x3f3f3f3f
using namespace std;
typedef struct
{
int x, y;
} pos;
vector<int> pre;
vector<int> erank;
vector<pos> v(1);
int find(int a)
{
if (pre[a] == a)
return a;
return pre[a] = find(pre[a]);
}
void join(int a, int b)
{
int x = find(a);
int y = find(b);
if (erank[x] < erank[y])
{
pre[x] = y;
}
else
{
if (erank[x] == erank[y])
{
erank[x]++;
}
pre[y] = x;
}
}
double getlen(pos x, pos y)
{
return sqrt((x.x - y.x) * (x.x - y.x) + (x.y - y.y) * (x.y - y.y));
}
int main()
{
int n, d;
scanf("%d %d", &n, &d);
bool vis[n + 1];
memset(vis, false, sizeof vis);
pre.resize(n + 1);
erank.resize(n + 1);
for (int i = 1; i <= n; i++)
{
pre[i] = i;
erank[i] = 1;
}
for (int i = 0; i < n; i++)
{
pos buf;
scanf("%d %d", &buf.x, &buf.y);
v.push_back(buf);
}
char c;
int p, q;
while (scanf("%c", &c) != EOF)
{
if (c == 'O')
{
scanf("%d", &p);
vis[p] = true;
for (int i = 1; i <= n; i++)
{
if (vis[i] == true && getlen(v[i], v[p]) <= d && i != p)
{
join(i, p);
}
}
}
else if (c == 'S')
{
scanf("%d %d", &p, &q);
if (find(p) == find(q))
{
printf("SUCCESS\n");
}
else
{
printf("FAIL\n");
}
}
// for (int i = 1; i <= n; i++)
// {
// printf("%d ", pre[i]);
// }
// printf("\n");
// for (int i = 1; i <= n; i++)
// {
// printf("%d ", erank[i]);
// }
// printf("\n");
}
return 0;
}