洛谷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是否具有亲戚关系。

输出格式

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

//对于100%的数据,N<=10000,M<=200000。
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1e4+5;
int father[maxn];
void init()
{
    for(int i=1;i<=maxn;++i)
    {
        father[i]=i;
    }
}
int find(int x)
{
    if(x==father[x])
    {
        return x;
    }
    return father[x]=find(father[x]);
}
void merge(int x,int y)
{
    int p = find(x);
    int q = find(y);
    if(p!=q)
    {
        father[p]=q;
    }
}
bool judge(int x,int y)
{
    int p = find(x);
    int q = find(y);
    if(p!=q)
    {
        return true;
    }
    else return false;
}
int main()
{
    int n,m,p;
    while(cin>>n>>m>>p)
    {
        init();
        int x,y;
        for(int i=1;i<=m;++i)
        {
            cin>>x>>y;
            merge(x,y);
        }
        for(int i=0;i<p;++i)
        {
            cin>>x>>y;
            if(judge(x,y))
            {
                cout<<"No"<<endl; 
            }else 
            {
                cout<<"Yes"<<endl;
            }
        }
    }
    return 0;
}
Last modification:January 12th, 2020 at 01:53 am
如果觉得我的文章对你有用,请随意赞赏