并查集
今天简略学习了一下并查集,大致思想就是将一个连通关系简化成找各自的前驱节点直到前驱为自己,最后会形成若干个树,对这些树状关系求解问题。
这篇讲的很细,mark一下。传送门
相关题目后续补上, 以下是模板
#include<bits/stdc++.h>
#define maxn 105
#define maxm 105
using namespace std;
int m,n;// n个点,m 条边
int pre[maxn]; //祖先
int h[maxn]; //树高
void init()//初始化,一定要有!
{
for(int i = 1;i <= n;i++){
h[i] = 1;//树高
pre[i] = i;//祖先最开始是它本身
}//初始化每一个点
return;
}
int find(int x)//找祖先的函数
{
if(pre[x] == x) return x;//如果祖先是他自己时返回自己
else{
pre[x] = find(pre[x]);//找到他的最终祖先
return pre[x];//顺便路径压缩,一层一层返回上去它的祖先
}
}
void merge(int a,int b)//将a和b合并到一棵树上面
{
int t1 = find(a),t2 = find(b);//分别找a,b的最终祖先
if(t1 == t2) return;//已经在同一棵树中
if(h[t1] < h[t2]) swap(t1, t2);//靠左原则,左边比右边的树要高
if(t1 != t2){
pre[t2] = t1;//左边的变成右边的祖先
h[t1] = max(h[t1], h[t2]+1);//合并树高,原树高与合并来的比较并进行更新
}
return;
}
int main()
{
cin >> n >> m;
init();
for(int i = 1;i <= m;i++)
{
int a, b;
cin >> a >> b;
merge(a, b);//合并a,b所在的树
}
// 。。。。。。//其他操作
return 0;
}