并查集 (Union-Find Sets)

http://lilongfei1030.blog.163.com/blog/static/860152820087201031473/

题目: 亲戚(Relations)

或许你并不知道,你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥的表姐的孙子。如果能得到完整的家谱,判断两个人是否亲戚应该是可行的,但如果两个人的最近公共祖先与他们相隔好几代,使得家谱十分庞大,那么检验亲戚关系实非人力所能及.在这种情况下,最好的帮手就是计算机。

为了将问题简化,你将得到一些亲戚关系的信息,如同MarryTom是亲戚,TomB en是亲戚,等等。从这些信息中,你可以推出MarryBen是亲戚。请写一个程序,对于我们的关心的亲戚关系的提问,以最快的速度给出答案。

参考输入输出格式 输入由两部分组成。

第一部分以NM开始。N为问题涉及的人的个数(1 N 20000)。这些人的编号为1,2,3,,N。下面有M(1 M 1000000),每行有两个数ai, bi,表示已知aibi是亲戚.

第二部分以Q开始。以下Q行有Q个询问(1 Q 1 000 000),每行为ci, di,表示询问cidi是否为亲戚。

对于每个询问ci, di,若cidi为亲戚,则输出Yes,否则输出No

样例输入与输出

输入relation.in

10 7

2 4

5 7

1 3

8 9

1 2

5 6

2 3

3

3 4

7 10

8 9

输出relation.out

Yes

No

Yes

 

 

#include<iostream>

using namespace std;

int N,M,Q;

int pre[20000],rank[20000];

void makeset(int x)

{

	pre[x]=-1;

	//rank[x]=0;

} 

int find(int x)

{

// 	int r=x;
// 
// 	while(pre[r]!=-1)
// 
// 		r=pre[r];
// 
// 	while(x!=r)//使路径上所有元素都指向根节点
// 
// 	{
// 
// 		int q=pre[x];
// 
// 		pre[x]=r;
// 
// 		x=q;
// 
// 	}
// 
// 	return r;       
	if (pre[x]<0)
	{
		return x;
	}
	else
		return pre[x]=find(pre[x]);

}     

void unionone(int a,int b)

{

	int t1=find(a);

	int t2=find(b);

	if(pre[t1]<pre[t2])

		pre[t2]=t1;

	else

		pre[t1]=t2;

	if(pre[t1]==pre[t2])

		pre[t2]--;      

}        

int main()



{ 

	int i,a,b,c,d;

	while(cin>>N>>M)

	{

		for(i=1;i<=N;i++)

			makeset(i);

		for(i=1;i<=M;i++)

		{

			cin>>a>>b;

			if(find(a)!=find(b))

				unionone(a,b);

		}

		cin>>Q;  

		for(i=1;i<=Q;i++)

		{

			cin>>c>>d;

			if(find(c)==find(d))

				cout<<"YES"<<endl;

			else

				cout<<"NO"<<endl;  

		}            

	}    

	return 0;

} 

  

posted @ 2011-10-17 14:47  refazy  阅读(288)  评论(0编辑  收藏  举报