kuangbin 并查集 - POJ - 2236 Wireless Network (并查集简单题)
[kuangbin带你飞] 题单 最短路问题 + 并查集问题
https://blog.csdn.net/m0_46272108/article/details/108923142
题意: 有一个计算机网络的所有线路都坏了,网络中有n台计算机,现在你可以做两种操作,修理(O)和检测两台计算机是否连通(S),只有修理好的计算机才能连通。连通有个规则,两台计算机的距离不能超过给定的最大距离D(一开始会给你n台计算机的坐标)。检测的时候输出两台计算机是否能连通。
题解: (详情请看代码~~~ ,代码注释写的真的很详细了。)
对并查集不太了解原理的可以看看:
https://blog.csdn.net/m0_46272108/article/details/108904480
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define ll long long
//#define int ll
#define INF 0x3f3f3f3f
using namespace std;
int read()
{
int w = 1, s = 0;
char ch = getchar();
while (ch < '0' || ch>'9') { if (ch == '-') w = -1; ch = getchar(); }
while (ch >= '0' && ch <= '9') { s = s * 10 + ch - '0'; ch = getchar(); }
return s * w;
//最大公约数
}int gcd(int x,int y) {
if(x<y) swap(x,y);//很多人会遗忘,大数在前小数在后
//递归终止条件千万不要漏了,辗转相除法
return x % y ? gcd(y, x % y) : y;
}
int lcm(int x,int y)//计算x和y的最小公倍数
{
return x * y / gcd(x, y);//使用公式
}
//------------------------ 以上是我常用模板与刷题几乎无关 ------------------------//
const int N = 1010;
double x[N], y[N];//电脑的坐标x,y
int good[N];//已经修好的电脑
int pre[N];//当前电脑的父节点
//n为电脑的数量,d为两台电脑之间能够直接通信的最大距离。
int n, d;
int cnt = 0;
//返回x的祖宗节点+路径压缩
int find(int x) {
//如果当前节点x不是根节点的话,就会往上走一层。
if (pre[x] != x) {
pre[x] = find(pre[x]);//就返回父节点的祖宗节点
}
return pre[x];
}
//集合合并
void join(int a, int b) {
//如果两个根节点不相同的话
if (find(a) != find(b)) {
//连通
pre[find(a)] = find(b);
}
}
//两台电脑之间的距离
double dis( int i, int j ) {
return sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]));
}
int main()
{
n = read(), d = read();
for (int i = 1; i <= n; ++i) {
cin >> x[i] >> y[i];
}
for (int i = 1; i <= n; ++i) {
pre[i] = i;
}
int a, b;
char op;
while(cin >> op){
//读入操作
if (op == 'O') {
a = read();
for(int i = 0; i < cnt; ++i) {
//如果两个点的距离小于d,则直接连通
if (dis(a, good[i]) <= d) {
join (a, good[i]);
}
}
//a是修好的电脑
good[cnt++] = a;
} else if (op = 'S') {
a = read(), b = read();
//判断a和b是否处于同一个连通块。
if (find(a) == find(b)) {
printf("SUCCESS\n");
} else {
printf("FAIL\n");
}
}
}
return 0;
}