题目描述
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。
Input
第一行:三个整数n,m,p,(n<=50000,m<=50000,p<=50000),分别表示有n个人,m个亲戚关系,询问p对亲戚关系。以下m行:每行两个数Mi,Mj,1<=Mi,Mj<=N,表示Ai和Bi具有亲戚关系。接下来p行:每行两个数Pi,Pj,询问Pi和Pj是否具有亲戚关系。
Output
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
分析
并查集模版。。
首先并查集两个函数打好,查找和合并都OK了以后
直接一条一条的读入,如果两个读入的亲戚不是一个树里的就合并。
然后查找的时候压缩路径就好了。。
#include <iostream>
#include <cstdio>
using namespace std;
int n,m,p,f[50001],i,x,y,r[50001];
int fin(int x)
{
int d=x,r,t;
while (f[d]>0)
d=f[d];
r=d;
d=x;
while (f[d]>0)
{
t=f[d];
f[d]=r;
d=t;
}
return r;
}
void uni(int x,int y)
{
int i=fin(x),j=fin(y);
if (r[i]<=r[j])
{
f[i]=j;
if (r[i]==r[j]) r[j]++;
return;
}
else
f[j]=i;
}
int main()
{
scanf("%d%d%d",&n,&m,&p);
for (i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
if (fin(x)!=fin(y)) uni(x,y);
}
for (i=1;i<=p;i++)
{
scanf("%d%d",&x,&y);
if (fin(x)==fin(y))
printf("Yes\n");
else
printf("No\n");
}
}