蓝桥杯朋友——C语言

朋友

在社交的过程中，通过朋友，也能认识新的朋友。在某个朋友关系图中，假定A和B是朋友，B和C是朋友，那么A和C也会成为朋友。即，我们规定朋友的朋友也是朋友。现在，已知若干对朋友关系，询问某两个人是不是朋友。请编写一个程序来解决这个问题吧。 输入格式：第一行：三个整数n、m、p（n,m,p≤5000）分别表示有n个人，m个朋友关系，询问p对朋友关系。接下来m行：每行两个数Ai,Bi（1≤Ai,Bi≤N），表示Ai和Bi具有朋友关系。接下来p行：每行两个数，询问两人是否为朋友。 输出格式：输出共p行，每行一个Yes或No。表示第iii个询问的答案为是否朋友。

样例输入
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

样例输出
YES
YES
NO

// 此题考的知识点是并查集
#include <stdio.h>
#define MAX 100001
int a[MAX];
int i;

// 初始化
void init(int x) {
	for (i = 1; i <= x; i++) {
		a[i] = i;
	}
}
 
// 查找 根节点 
int find(int x) {
	if (x == a[x])
		return x;
	else {
		a[x] = find(a[x]);
		return a[x];
	}
}


// 合并根节点 
void unionn(int i, int j) {
	int i_a = find(i);
	int j_a = find(j);
	a[i_a] = j_a; 
}



main() {
	int n, m, p, Ai, Bi;
	scanf("%d%d%d", &n, &m, &p);
	init(n); 
	for (i = 1; i <= m; i++) {
		scanf("%d%d", &Ai, &Bi);
		unionn(Ai, Bi);
	} 
	
	for (i = 1; i <= p; i++) {
		scanf("%d%d", &Ai, &Bi);
		if (find(Ai) == find(Bi)) 
			printf("YES\n");
		else 
			printf("NO\n");
	}
}