并查集(Disjoint Set Union, DSU)是一种用于解决集合合并与查询问题的数据结构。它维护一组不相交的动态集合,支持以下两种操作:
1. 查找(Find):确定一个元素属于哪个集合,即找到代表元素(也称为根节点)。
2. 合并(Union):将两个集合合并成一个集合,即将根节点合并。
并查集通过使用数组或树来实现,在树的表示中,每个集合由一棵树表示,根节点表示集合的代表元素。并查集的关键操作是查找和合并。
查找操作的实现方法为:从给定元素开始,通过不断向上寻找父节点,直到找到根节点,即找到集合的代表元素。
合并操作的实现方法为:将两个集合的根节点连接在一起,可以通过将其中一个根节点的父节点指向另一个根节点实现。
并查集可以求连通子图,最小生成树Kruskal算法和公共祖先(LCA)等等;
基本操作:
1.初始化init;
2.查询find;
3.合并unionn;
#include <bits/stdc++.h>
#define ll long long
#define endl "\n"
#define KUI ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
using namespace std;
const int con = 2e5 + 4;
const ll mod = 998244353;
int n, m, k;
int fa[con];
// 查询根祖先;
int findfa(int i)
{
if (i == fa[i])
{
return i; // 找到并返回祖先位置;
}
else
{
fa[i] = findfa(fa[i]); // 路径压缩,关键的一步
// 返回找到的根祖先设为每一个子节点的祖先,大大增加了搜索的效率;
return fa[i];
}
}
// 合并祖先;
void hebing(int i, int j)
{
int ifa = findfa(i); // 找到i的祖先
int jfa = findfa(j); // 找到j的祖先
fa[ifa] = jfa; // 将i的祖先指向j;
}
int main()
{
KUI;
cin >> n >> m;
// 一开始,我们把父节点设为数本身;
for (int i = 1; i <= n; i++)
{
fa[i] = i;
}
int ans1, ans2;
for (int i = 1; i <= m; i++)
{
cin >> ans1 >> ans2;
hebing(ans1, ans2);
}
cin >> k;
for (int i = 1; i <= k; i++)
{
cin >> ans1 >> ans2;
if (findfa(ans1) == findfa(ans2))
{
cout << "yes" << endl;
}
else
{
cout << "no" << endl;
};
}
return 0;
}