1073 家族（并查集模板）-蒲公英云

1073 家族（并查集模板）

题目描述 Description

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。规定：x和y是亲戚，y和z是亲戚，那么x和z也是亲戚。如果x,y是亲戚，那么x的亲戚都是y的亲戚，y的亲戚也都是x的亲戚。

输入描述 Input Description

第一行：三个整数n,m,p，（n<=5000,m<=5000,p<=5000），分别表示有n个人，m个亲戚关系，询问p对亲戚关系。以下m行：每行两个数Mi，Mj，1<=Mi，Mj<=N，表示Ai和Bi具有亲戚关系。接下来p行：每行两个数Pi，Pj，询问Pi和Pj是否具有亲戚关系。

输出描述 Output Description

P行，每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。

样例输入 Sample Input

6 5 3

1 2

1 5

3 4

5 2

1 3

1 4

2 3

5 6

样例输出 Sample Output

Yes

数据范围及提示 Data Size & Hint

n<=5000,m<=5000,p<=5000

#include<iostream>
#include<string.h>
#include<cstdio> 
using namespace std;
int pre[30000];
int n,m,p;
void init()
{
    for (int i=1; i<=n; i++)
    {
        pre[i] = -1; 
    } 
}
int find(int num)
{
    while (pre[num]>-1)
    {
        num = pre[num];
    }
    return num;
}
void union_set(int x, int y)
{
    int xx = find(x);
    int yy = find(y);
    if (xx != yy)
    {
        pre[xx] += pre[yy];
        pre[yy] = xx;
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&p); 
    init();
    int x, y;
    while (m--)
    {
        scanf("%d%d",&x,&y);
        union_set(x,y);
    }
    while (p--)
    {
        scanf("%d%d",&x,&y);
         if (find(x) == find(y))
         {
             cout<<"Yes"<<endl;
         }else{
             cout<<"No"<<endl; 
         }
    }
    return 0;
}