洛谷P1551 亲戚

题目背景

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。

题目描述

规定：x和y是亲戚，y和z是亲戚，那么x和z也是亲戚。如果x,y是亲戚，那么x的亲戚都是y的亲戚，y的亲戚也都是x的亲戚。

输入格式

第一行：三个整数n,m,p，（n<=5000,m<=5000,p<=5000），分别表示有n个人，m个亲戚关系，询问p对亲戚关系。

以下m行：每行两个数Mi，Mj，1<=Mi，Mj<=N，表示Mi和Mj具有亲戚关系。

接下来p行：每行两个数Pi，Pj，询问Pi和Pj是否具有亲戚关系。

题解：并查集模板题。

#include<cstdio>
#include<iostream>

using namespace std;

int f[5005],n,m,k;

int find(int a)
{
    if(f[a]==a) return f[a];
    return f[a]=find(f[a]);
}

void merge(int a,int b)
{
    f[a]=b;
    return;
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++)
        f[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        int f1=find(x);
        int f2=find(y);
        if(f1==f2) continue;
        else merge(f1,f2);
    }
    for(int i=1;i<=k;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        int f1=find(x);
        int f2=find(y);
        if(f1==f2) printf("Yes
");
        else printf("No
"); 
    }
    return 0;
}