蓝桥杯之每日一题

解题思路
并查集模板题

解题代码

#include<iostream>
using namespace std;
const int N = 5005;

int n, m, p;
int a[N], rk[N];//rk记录当前根节点在树中的深度 
//并查集
//初始化 
void init(int n) {
	for (int i = 1; i <= n; i++) {
		a[i] = i;
		rk[i] = i;
	}
}

//查询
int find(int x) {
	if (x == a[x]) {
		return x;//若待查询元素就是祖先元素 
	}
	else {
		return find(a[x]);//若不是，返回上一层查找 
	}
}
//查询（路径压缩）
//路径压缩可以使每个节点到根节点的路径最短，提高查询效率，但是路径压缩和按秩合并一起用，可能会破坏数组rk的准确性 
//int find(int x) {
//	if(x == a[x]){
//		return x;
//	}
//	else{
//		a[x] == find(a[x]);
//		return a[x];
//	}
//}
//合并
//void merge(int x, int y){
//	a[find(x) = a[find(y)];
//}
//合并 (按秩合并)
//因为路径压缩仅能压缩当前查询的路径，按秩合并可以压缩未查询的路径，但是两者不能同时使用
void merge(int x, int y) {
	int ra = find(x), rb = find(y);
	if (rk[ra] <= rk[rb]) {
		a[ra] = rb;//深度小的树向深度大的树合并 
	}
	else {
		a[rb] = ra;
	}
	//若两树深度相同且根节点不同，新的根节点深度+1，两树已在第一个if处合并 
	if (rk[ra] == rk[rb] && ra != rb) {
		rk[rb]++;
	}
}

int main()
{
	cin >> n >> m >> p;
	init(n);
	int x, y;
	//输入一个关系合并一次 
	while (m--) {
		cin >> x >> y;
		merge(x, y);
	}
	// 查询 
	while (p--) {
		cin >> x >> y;
		if (find(x) == find(y))
			cout << "Yes" << endl;
		else
			cout << "No" << endl;
	}

	return 0;
}