洛谷 P3367 【模板】并查集

并查集顾名思义，并是合并的意思，查是查找的意思，集是集合的意思。

并查集定义：1.并查集是一种树的数据结构，用于处理一些不相交集合的合并以及查询是否在一个集合的操作。2.并查集通常包含两种操作：查找（find）函数用来查询两个元素是否在同一集合当中；合并（union）函数用来将两个元素合并为同一集合。

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入格式

第一行包含两个整数 N,M ,表示共有 N 个元素和 M 个操作。

接下来 M 行，每行包含三个整数 Zi​,Xi​,Yi​ 。

当 Zi​=1 时，将Xi​ 与Yi​ 所在的集合合并。

当 Zi​=2 时，输出Xi​ 与Yi​ 是否在同一集合内，是的输出 Y ；否则输出 N 。

输出格式

对于每一个Zi​=2 的操作，都有一行输出，每行包含一个大写字母，为 Y 或者 N 。

输入输出样例

输入 #1复制

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

输出 #1复制

N
Y
N
Y

说明/提示

对于 30% 的数据，N≤10，M≤20。

对于 70% 的数据，N≤100，M≤10^3。

对于100% 的数据，1≤N≤10^4，1≤M≤2×10^5，1≤Xi​,Yi​≤N，Zi​∈{1,2}。

int find(int k) { //找老大函数
	//中间有个递归实现压缩路径减少时间
	//我们要从他这个位置开始往上查找知道第一个根节点，在路过他所有上级的过程中，我们可以顺便途中经过的人的大哥替换为所有人的大哥
	if (f[k] == k)
		return k;//返回老大是谁
	else
		return f[k] = find(f[k]); //递归
}

查找（find）函数主要的功能是查找两个集合的父亲结点，判断是否相等，若相等则说明两个元素在一起若果不相等，说明两个元素不在一个集合内。细节的地方就是包含一个递归算法，实现压缩路径减少时间防止TLE。在查找父亲结点过程中，把其中路过的所有元素的父亲结点替换为所有元素的父亲结点，减少时间消耗。

if (z[i] == 1) { //合并xi和yi
			f[find(y[i])] = find(x[i]); //xi变成yi的老大
		} else { //判断是否是一伙的
			if (find(x[i]) == find(y[i]))
				printf("Y\n");
			else
				printf("N\n");
		}

合并函数，主要是通过改变两个集合的父亲节点，父亲节点相同则说明了两个元素在同一集合内。

下面附上完整代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 200005;
int n, m;
int z[N], x[N], y[N];
int f[10005];//用f[i]表示第i个人的老大是谁
int find(int k) { //找老大函数
	//中间有个递归实现压缩路径减少时间
	//我们要从他这个位置开始往上查找知道第一个根节点，在路过他所有上级的过程中，我们可以顺便途中经过的人的大哥替换为所有人的大哥
	if (f[k] == k)
		return k;//返回老大是谁
	else
		return f[k] = find(f[k]); //递归
}
int main() {
	scanf("%d %d", &n, &m);
	for (int i = 1; i <= n; i++)
		f[i] = i; //刚开始自己单独一个集合，所以老大就是自己（赋初值）
	for (int i = 1; i <= m; i++) {
		scanf("%d %d %d", &z[i], &x[i], &y[i]);
		if (z[i] == 1) { //合并xi和yi
			f[find(y[i])] = find(x[i]); //xi变成yi的老大
		} else { //判断是否是一伙的
			if (find(x[i]) == find(y[i]))
				printf("Y\n");
			else
				printf("N\n");
		}
	}
	return 0;
}