POJ 2236 Wireless Network

最新推荐文章于 2024-06-14 20:42:49 发布

原创最新推荐文章于 2024-06-14 20:42:49 发布 · 631 阅读

0 ·

CC 4.0 BY-SA版权

1下暑假集训专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种使用并查集解决通信网络连通性的算法。通过路径压缩和按秩合并的方法来高效判断两个节点是否属于同一集合，进而确定它们是否能够进行通信。文章提供了完整的C++代码实现。

思路详见课本 P 213

思路：直接用并查集，最后看 p 和 q 是否在一个集合中即可。属于同一集合，则可以通信；否则失败。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=1000 +100;
int set[maxn];
int xx[maxn];
int yy[maxn];
bool valid[maxn];
int n,d;

int set_find(int d)//路径压缩的 合并树根 的函数
{
	if(set[d]<0)
		return d;
	return set[d]=set_find(set[d]);
}
void join(int x,int y)
{
	x=set_find(x);
	y=set_find(y);
	if(x!=y)//-----------此处这个判断千万不能少，否则在 set_find()函数中，出现死循环（由于节点 自己 指向 自己，找不到 结束条件 （即树根））
		//------------提交会出现runtime error  
		set[x]=y;
}
int in(int i,int p)//判断i 是否在 p的范围内
{
	if(  (xx[i]-xx[p]) * (xx[i]-xx[p]) + (yy[i]-yy[p]) * (yy[i]-yy[p]) <= d*d  )
		return 1;
	return 0;
}
void repair(int p)//修复 p
{
	valid[p]=1;
	int i;
	for(i=1;i<=n;i++) 
		if( valid[i] &&in(i,p) &&i!=p)
			join(i,p);
}
void search(int p,int q) //查看 p  和 q是否 可以 通信
{
	if( set_find(p)==set_find(q) )
		cout<<"SUCCESS"<<endl;
	else
		cout<<"FAIL"<<endl;
}
int main()
{
	memset(set,-1,sizeof(set));
	memset(xx,0,sizeof(xx));
	memset(yy,0,sizeof(yy));
	memset(valid,0,sizeof(valid));
	cin>>n>>d;
	int i;
	for(i=1;i<=n;i++)
		cin>>xx[i]>>yy[i];
	char c;
	while(cin>>c)
	{
		if(c=='O')
		{
			int p;
			cin>>p;
			repair(p);
		}
		else if(c=='S')
		{
			int p,q;
			cin>>p>>q;
			search(p,q);
		}
	}
	return 0;
}