并查集入门

1.引入

大家在奋力码题的时候，总是会有一些题需要你合并一些东东，常规的数据结构无法

快速合并并查询，这时候就要用到并查集了。

例如：   冰茶寄（并查集）模板

2.并查集的模板

#include <iostream>
#include <vector>
using namespace std;

// 定义并查集类
class DisjointSet {
    private:
        vector<int> parent;
        vector<int> rank;
    
    public:
        // 构造函数，初始化并查集
        DisjointSet(int n) {
            parent.resize(n);
            rank.resize(n, 0);
            for (int i = 0; i < n; ++i) {
                parent[i] = i;
            }
        }
        
        // 查找根节点
        int find(int x) {
            return parent[x]==x?x:find(parent[x]);
        }
        
        // 合并两个集合
        void unite(int x, int y) {
            int rootX = find(x);
            int rootY = find(y);
            
            if (rootX == rootY) {
                return;
            }
            
            if (rank[rootX] < rank[rootY]) {
                parent[rootX] = rootY;
            } else if (rank[rootX] > rank[rootY]) {
                parent[rootY] = rootX;
            } else {
                parent[rootX] = rootY;
                rank[rootY]++;
            }
        }
};

int main() {
    int n = 5; // 元素个数
    DisjointSet ds(n);
    
    ds.unite(0, 1); // 合并元素0和1所在的集合
    ds.unite(2, 3); // 合并元素2和3所在的集合
    
    cout << ds.find(1) << endl; // 查询元素1所在的集合的根节点
    
    return 0;
}
//作者太懒了，这个代码不是他写的

我们用parent[x]记录x的最近祖先，一旦x是没有祖先的那么parent[x]=x;

要想找到x的最早祖先就要不断查询直到parent[x]=x

合并两个并查集就要用其中一个的最早祖先的parent[x]指向另一个冰茶寄的最早祖先

3.优化

如果你学到这里就兴冲冲的提交那道题，恭喜你70ptsTLE;

为什么呢，因为你每次查询都要花最多O(N)的时间，所以会超时. 

所以我们要路径压缩！

上面的并查集相当于一棵树，而且是深度远大于直径的二叉树！

这时候查询就要很长时间，所以我们将所有结点指向根节点，这样查询只需要查一次

就可以了

AC Code:

#include<bits/stdc++.h>
using namespace std;
int n,m,s[10002];
void init_set(){
	for(int i=1;i<=n;++i) s[i]=i;//初始化
}
int find_set(int x){
	if(s[x]==x) return x;
	else return s[x]=find_set(s[x]);//路径压缩
}
void hbng_set(int x,int y){
	s[find_set(x)]=find_set(y);//合并
}
int main(){
	cin>>n>>m;
	init_set();
	for(int i=1;i<=m;++i){
		int op,x,y;
		cin>>op>>x>>y;
		if(op==1)
			hbng_set(x,y);
		else{
			int nx=find_set(x),ny=find_set(y);
			if(nx==ny) cout<<"Y"<<endl;
			else cout<<"N"<<endl;
		}
	}
	return 0;
}

谢谢