例题
找亲戚
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。 规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。
Input
第一行:三个整数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是否具有亲戚关系。
Output
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
int father[50001],a,b,m,n,p;
int find(int x){
if (father[x]!=x) father[x]=find(father[x]);
return father[x];
}
int main() {
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) father[i]=i;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&a,&b);
a=find(a);
b=find(b);
father[b]=a;//路径压缩
}
scanf("%d",&p);
for (int i=1;i<=p;i++){
scanf("%d%d",&a,&b);
if (find(a)==find(b)) printf("YES\n");
else printf("NO\n");
}
return 0;
}
在判断两个人是不是亲戚时,不能简单地判断是否father[a]==father[b],应为此时的father数组只是进行了一次路径压缩,不能用来判断是否具有相同祖先。所以需要在查找时再次find(a)、find(b)此时才能再次把要找的点进行路径压缩。