并查集是一种树形结构,经常用于处理一些不相交集合的查询和合并,并且时间复杂度近乎O(1)。
板子题
题意:
给一些编号从1到n的元素,每个元素一开始独自在一个集合中,然后可以对这些元素进行如下的操作:
1. 给出两个元素编号a和b输出两个元素是否在一个集合中
2. 给出两个元素编号a和b将两个元素所在的集合合并
思路:
如果使用普通的做法会如何? 我们可以用一个数组来储存哪个元素在哪个集合,执行操作一时只需要看两个元素所在的集合号是否相同即可。
但是合并操作呢?我们就需要把一个集合中所有元素的编号改成另个一集合的编号了显然很费时间。
此时我们就可以用并查集,我们把每一个集合都看成一棵树,并且规定:
- 每一个节点都储存它的父节点
- 祖宗节点储存他本身
这样啊的话我们就可以很好地解决这个问题了,每次我们查找两个元素是否在同一集合时就可以顺着他们节点的父亲,查找他们是否是有共同的祖宗。
那么合并操作呢?非常简单,只需要把一个集合的祖宗节点指向另一个集合的祖宗节点就好了,因为这样两个集合的元素就有共同的祖宗了。
路径压缩优化:
对于查寻一个节点的祖宗节点的时间,明显和树的深度有关,如果树的深度过大那么就会很费时间,此时就可以在每次查询时,把路径上所有的节点都指向祖宗节点。
代码:
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll maxn = 1e5 + 5;
ll n,m,p[maxn];
// find函数用于查询一个节点的祖宗
int find(int x)
{
if(p[x] != x)
p[x] = finds(p[x]);// 把路径上的所有点都指向祖宗
return p[x];
}
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++)
p[i] = i;//最开始一个节点一棵树
while(m--)
{
char c;
ll a,b;
cin >> c >> a >> b;
if(c == 'Q')
{
if(find(a) == find(b))
cout << "Yes" << endl;
else
cout << "No" << endl;
}
else
p[find(a)] = find(b);
//集合合并
}
return 0;
}