并查集 PID331 / 家族
题目描述
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
规定: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,表示Ai和Bi具有亲戚关系。
接下来p行:每行两个数Pi,Pj,询问Pi和Pj是否具有亲戚关系。
输出格式
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
样例输入
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
样例输出
Yes
Yes
No#include<iostream>
#include<stack> #include<queue> #include<cstring> #define maxn 50005 using namespace std; int a[maxn]={0}; int find(int n) { if(a[n]==n) { return n; } a[n]=find(a[n]); return a[n]; } void union1(int c,int d) { int fa=find(c); int fb=find(d); a[fb]= fa; } int main() { int n,m,p,a1,a2,b1,b2; cin>>n>>m>>p; for(int i=0;i<maxn;i++) a[i]=i; while(m--) { cin>>a1>>a2; union1(a1,a2); } while(p--) { cin>>b1>>b2; if(find(b1)==find(b2)) cout<<"Yes"<<endl; else cout<<"No"<<endl; } return 0; }
对了一开始我总有invalid types 'int[int]' for array subscript这个错误
int a[maxn]={0};
int find(int n)
{
if(a[n]==n)
{
return n;
}
a[n]=find(a[n]);
return a[n];
}
void union1(int a,int b)
{
int fa=find(a);
int fb=find(b);
a[fb]= fa;
}
后经高人指点才知道
void union1(int a,int b)
{
int fa=find(a);
int fb=find(b);
a[fb]= fa;
}
看到 a 的定义了吧,它是 int a; 而非 int a[maxn]={0};
如果你够坚强够勇敢,你就能驾驭他们