Wireless Network 并查集
题目大意:
有几个点,他们之间不超过一定距离的话,可以联通,
然后有好多点,都是坏的,有两种行为一个是选择修复某一个点,另一个是测试两个点之间是否可以联通,如果可以联通的话,就输出SUCCESS 否则的话就输出FAIL
其中,任意一个点都可以作为中转站
解体思路:
每修复一个点,就看下它与其他已经修复的点直接是否能联通,能得话就联通,
然后构建一个树,判断是否联通就看下根节点是否为同一个根节点
代码:
#include <iostream>
#include <cstring>
struct Greed //node节点
{
double x, y;
}dir[1111];
int par[1111]; //树
int ok[1111]; //已经修复的点
int rank[1111]; //防止退化
bool vj(Greed a, Greed b, int d) //判断两个点之间是否能联通
{
if((a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y) <= d * d) return true;
return false;
}
int find(int a) //寻找这个点的根节点
{
if(a == par[a]) return a;
else return par[a] = find(par[a]);
}
void link(int a, int b) //连接两个点
{
a = find(a);
b = find(b);
if(rank[a] < rank[b]) par[a] = b;
else par[b] = a;
if(rank[a] == rank[b]) rank[a]++; //防止树的高度过长
}
int main()
{
int n, d;
while(std::cin >> n >> d) //d为最长联通距离
{
int cnt = 0;
memset(rank, 0, sizeof rank);
for(int i = 1; i < 1111; ++i) par[i] = i; //初始化树
for(int i = 0; i < n; ++i)
std::cin >> dir[i+1].x >> dir[i+1].y;
char ch;
while(std::cin >> ch)
{
if(ch == 'O') //如果是O,就表示这个点修复了
{
int temp;
std::cin >> temp;
for(int i = 0; i < cnt; ++i)
{
if(vj(dir[temp], dir[ok[i]], d)) //link 修复的点之间,如果可以联通就联通,构建树
{
link(temp, ok[i]);
}
}
ok[cnt++] = temp; //将已经修复的点放数组里
}
else //如果是测试的话
{
int a, b;
std::cin >> a >> b;
//因为任何一个点都可以作为中间节点,所以只需要判断是否有同一根就好了
if(find(a) == find(b)) std::cout << "SUCCESS" << std::endl;
else std::cout << "FAIL" << std::endl;
}
}
}
return 0;
}