洛谷 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个询问的答案为“具有”或“不具有”亲戚关系。

输入输出样例

输入 #1 复制

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

输出 #1 复制

Yes
Yes
No



代码

#include<bits/stdc++.h>
using namespace std;
const int range=10005;
int fa[range];

int find(int x) {
 if(fa[x]!=x)
  fa[x]=find(fa[x]);
 return fa[x];
}

void merge(int x,int y) {
 int fx=find(x),fy=find(y);
 if(fx!=fy)   fa[fy]=fx;
}

int main() {
 int n,m,p;
 scanf("%d%d%d",&n,&m,&p);
 for(int i=1; i<=n; i++)
  fa[i]=i;
 int x,y;
 while(m--) {
  scanf("%d%d",&x,&y);
  fa[find(x)]=find(y);
 }
 while(p--) {
  scanf("%d%d",&x,&y);
  if(find(x)==find(y)) printf("Yes ");
  else printf("No ");
 }
 return 0;

}