并查集——亲戚(relation)

问题 N: 【例4-7】亲戚(relation)

时间限制: 1 Sec  内存限制: 128 MB
提交: 126  解决: 76
[提交][状态][讨论版][命题人:quanxing]

题目描述

或许你并不知道，你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱，判断两个人是否是亲戚应该是可行的，但如果两个人的最近公共祖先与他们相隔好几代，使得家谱十分庞大，那么检验亲戚关系实非人力所能及。在这种情况下，最好的帮手就是计算机。为了将问题简化，你将得到一些亲戚关系的信息，如Marry和Tom是亲戚，Tom和Ben是亲戚，等等。从这些信息中，你可以推出Marry和Ben是亲戚。请写一个程序，对于我们的关于亲戚关系的提问，以最快的速度给出答案。

输入

输入由两部分组成。

第一部分以N，M开始。N为问题涉及的人的个数(1≤N≤20000)。这些人的编号为1,2,3,…, N。下面有M行(1≤M≤1000000)，每行有两个数ai,bi

，表示已知ai 和bi

是亲戚。

第二部分以Q开始。以下Q行有Q个询问(1≤ Q ≤1000000)，每行为ci,di

，表示询问ci 和di

是否为亲戚。

测试数据比较水

输出

对于每个询问ci,di

，输出一行：若ci 和di

为亲戚，则输出“Yes”，否则输出“No”。

样例输入

10 7
2 4
5 7
1 3
8 9
1 2
5 6
2 3
3
3 4
7 10
8 9

样例输出

Yes
No
Yes

#include<iostream>
using namespace std;
int n,m,q,pre[20001],ai,bi,ci,di;
//找爹递归函数
int find(int x)
{
    int r=x;
    while(pre[r]!=r)
    r=pre[r];
    int a=x,b;
    while(a!=r)//路径压缩
    {
        b=pre[a];
        pre[a]=r;
        a=b;
    }
    return r;
}
//合并两个子集合
void join(int x,int y)
{
    int xp=find(x),yp=find(y);
    if(xp!=yp)
    pre[xp]=yp;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)//初始化只有一个节点的树
    pre[i]=i;
    for(int i=0;i<m;i++)
    {
        cin>>ai>>bi;
        join(ai,bi);
    }
    cin>>q;
    for(int i=0;i<q;i++)
    {
    cin>>ci>>di;
    if(find(ci)==find(di))cout<<"Yes"<<endl;
    else cout<<"No"<<endl;  
    }
    return 0;
}