一、并查集
主要用于处理一些不相交集合的合并问题,一些常见的用途有求连通子图、求最小生成树的Kruskal算法和求最近公共祖先(LCA)等
并查集的基本操作
1.初始化
假如有编号为1,2,3,…,n的n个元素,我们用一个数组fa[]来存储每个元素的父节点。一开始,我们将它们的父节点设为自己
void init(int n){
for(int i = 1; i <=n; i++){
fa[i] = i;
}
}
2.查询
找到i的祖先直接返回
分为路径压缩和路径未压缩
路径未压缩
int find(int i){ //未进行路径压缩
if(fa[i] == i){
return i;
}else{
return find(fa[i]);
}
}
路径压缩(优化版更常用)
int find(int i){ //采用路径压缩
if(fa[i] == i){
return i;
}else{
fa[i] = find(fa[i]); //进行路径压缩
return fa[i]; //返回父结点
}
}
3.合并
void unionn(int i,int j){
int i_fa = find(i); //找到i的祖先;
int j_fa = find(j); //找到j的祖先;
fa[i_fa] = j_fa; //将i的祖先指向j的祖先
}
如(4,5)的合并,将4的父亲指向5;
完整代码如下:
并查集代码:
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e6 + 10;
int fa[MAXN];
void init(int n){
for(int i = 1; i <=n; i++){
fa[i] = i;
}
}
// int find(int i){ //未进行路径压缩
// if(fa[i] == i){
// return i;
// }else{
// return find(fa[i]);
// }
// }
int find(int i){ //采用路径压缩
if(fa[i] == i){
return i;
}else{
fa[i] = find(fa[i]); //进行路径压缩
return fa[i]; //返回父结点
}
}
void unionn(int i,int j){
int i_fa = find(i);
int j_fa = find(j);
fa[i_fa] = j_fa;
}
int main(){}
使用以上优化算法虽然能加快查询时间,但是当数据很大时,有可能会有栈溢出的情况,于是提出一种优化查询的算法这种写法避免了回溯过程中返回值的问题,先找到父亲节点,在沿着路径进行一一修改。
合并操作也很简单。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e6 + 10;
int fa[MAXN];
void init(int n){
for(int i = 1; i <= n; i++){
fa[i] = i;
}
}
int find(int x){
int y = x;
while(y != fa[y]){
y = fa[y];
}
while(x != fa[x]){
int px = fa[x];
fa[x] = y;
x = px;
}
return y;
}
void unionn(int a,int b){
fa[b] = a;
}
int main(){
}
例题:
现在有若干家族关系,给出了一些亲戚关系,如Marry和Tom是亲戚,Tom和Ben是亲戚等。从这些信息中,你可以推导出Marry和Ben是亲戚。请写一个程序,对于我们的关于亲戚关系的提问,一最快速度给出答案。
输入格式: 第一部分是一M,N开始,N为人数(1<=N<=20000),这些人的编号为1,2,3,…,N。下面有M行(1<=M<=1000000),每行有两个数a,b,表示a和b是亲戚; 第二部分是以Q开始。以下Q行有Q个询问(1<=Q<=1000000),每行为c,表示询问c和d是否为亲戚。
输出格式: 对于询问c,d,输出一行:若c,d为亲戚,则输出"YES"否则输出"NO";
样例输入: 10 7 2 4 5 7 1 3 8 9 1 2 5 6 2 3 3 3 4 7 10 8 9
样例输出:
YES
NO
YES
#include <bits/stdc++.h>
using namespace std;
int fa[20001];
int init(int n)
{
for (int i = 1; i <= n; i++)
{
fa[i] = i;
}
}
int find(int i)
{
if (fa[i] == i)
{
return i;
}
else
{
fa[i] = find(fa[i]);
return fa[i];
}
}
void unionn(int i, int j)
{
int i_fa = find(i);
int j_fa = find(j);
fa[i_fa] = j_fa;
}
int main()
{
int n, m;
cin >> n >>m;init(n);
int a, b;
for (int i = 1; i <= m; i++)
{
cin >> a >> b;
unionn(a, b);
}
int q;
cin >> q;
for (int j = 1; j <= q; j++)
{
cin >> a >> b;
if (fa[a] == fa[b])
cout << "YES" << endl;
else
cout << "NO" << endl;
}
}