朋友 并查集

在社交的过程中，通过朋友，也能认识新的朋友。在某个朋友关系图中，假定 A 和 B 是朋友，B 和 C 是朋友，那么 A 和 C 也会成为朋友。即，我们规定朋友的朋友也是朋友。

现在，已知若干对朋友关系，询问某两个人是不是朋友。

请编写一个程序来解决这个问题吧。

输入格式

第一行：三个整数 $n,m,p(n\le 5000,m\le 5000,p\le 5000)$，分别表示有 $n$ 个人，$m$ 个朋友关系，询问 $p$ 对朋友关系。

接下来 $m$ 行：每行两个数 A_i,B_iAi​,Bi​，1\leq A_i,B_i\leq N1≤Ai​,Bi​≤N，表示 A_iAi​ 和 B_iBi​ 具有朋友关系。

接下来 $p$ 行：每行两个数，询问两人是否为朋友。

输出格式

输出共 $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 <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
#include <stack>
#include <set>
#include <queue>
using namespace std; 
const int inf=0x3f3f3f3f;

int n,m,p; 
int f[5005];

int find(int k){
	return f[k]==k?k:find(f[k]);
}

void merge(int u,int v){
	int t1=find(u);
	int t2=find(v);
	if(t1!=t2){
		f[t2]=t1;
	}
}

int main()
{   
	scanf("%d%d%d",&n,&m,&p);
	for(int i=1;i<=n;i++)
		f[i]=i;
	for(int i=1;i<=m;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		merge(u,v);
	}
	for(int i=1;i<=p;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		if(find(u)==find(v))	printf("Yes\n");
		else printf("No\n");
	}
	return 0;  
}
/*
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
*/