并查集
一、解决的问题
并查集被很多OIer认为是最简洁而优雅的数据结构之一,主要用于解决一些元素分组的问题。它管理一系列不相交的集合,并支持两种操作:
合并(Union):把两个不相交的集合合并为一个集合。
查询(Find):查询两个元素是否在同一个集合中。
代表某个节点与其父节点之间存在联系,它多用来判断图的连通性
二、常规并查集
三、路径压缩并查集
就是把每个元素都直接接在集合的根节点里
四、按秩合并(最优化的)
五、查找方法,判断两个元素是否同个根节点
六、步骤
1、初始化,把每个元素的根节点设置为自己
秩初始化都为1
void init(int n)
{
for (int i = 1; i <= n; ++i)
{
fa[i] = i;
rank[i] = 1;
}
}
2、查找,查找到两个元素的根节点(路径压缩)
int find(int x)
{
if(x == fa[x])
return x;
else{
fa[x] = find(fa[x]); //父节点设为根节点
return fa[x]; //返回父节点
}
}
3、合并,秩少的集合合并到秩多的集合(按秩合并)
void merge(int i, int j)
{
int x = find(i), y = find(j); //先找到两个根节点
if (rank[x] <= rank[y])
fa[x] = y;
else
fa[y] = x;
if (rank[x] == rank[y] && x != y)
rank[y]++; //如果深度相同且根节点不同,则新的根节点的深度+1
}
七、亲戚题目代码
#include<stdio.h>
int fa[5005];
int rank[5005];
void init(int n){
int i;
for(i=1;i<=n;i++){
fa[i]=i;
rank[i]=1;
}
}
int find(int x){
if(fa[x]==x)
return x;
else{
fa[x]=find(fa[x]);
return fa[x];
}
}
void merge(int i,int j){
int x,y;
x=find(i);
y=find(j);
if(rank[x]<=rank[y])
fa[x]=y;
else
fa[y]=x;
if(rank[x]==rank[y]&&x!=y)
rank[y]++;
}
int main(){
int n,m,p;
int k;
int i,j;
scanf("%d%d%d",&n,&m,&p);
init(n);
for(k=0;k<m;k++){
scanf("%d%d",&i,&j);
merge(i,j);
}
for(k=0;k<p;k++){
scanf("%d%d",&i,&j);
if(find(i)==find(j))
printf("Yes\n");
else
printf("No\n");
}
}
参考资料
算法学习笔记(1) : 并查集
带权并查集
一、作用
互通节点之间有数据
二、例题:网络分析
1、解释
在这里插入图片描述
2、代码
/*
带权并查集
*/
#include<stdio.h>
int dis[100005]; //与父节点差值
int value[100005]; //父节点的信息量
int fa[100005]; //父节点
int rank[100005]; //秩大小
//初始化
void init(int n){
int i;
for(i=1;i<n;i++){
fa[i]=i;
rank[i]=1;
value[i]=0;
dis[i]=0;
}
}
//查找父节点
int find(int x){
if(fa[x]==x)
return x;
else{
int temp=fa[x];
fa[x]=find(fa[x]);
//换fa【x】的父节点,故要加fa【x】的差值
dis[x]+=dis[temp];
return fa[x];
}
}
//合并元素
void merge(int i,int j){
int x,y;
x=find(i);
y=find(j);
if(rank[x]<=rank[y]){
fa[x]=y;
//两个节点的差值
dis[x]=value[x]-value[y];
}else{
fa[y]=x;
dis[y]=value[y]-value[x];
}
if(rank[x]=rank[y]&&x!=y){
rank[y]++;
}
}
int main(){
int n,m;
int k;
int i,j;
int flat;
scanf("%d%d",&n,&m);
init(n);
for(k=0;k<m;k++){
scanf("%d",&flat);
//合并元素
if(flat==1){
scanf("%d%d",&i,&j);
merge(i,j);
}
//发送信息
else{
scanf("%d%d",&i,&j);
//数据记录在根节点,这样每个节点的信息量等于=与父节点的差值+父节点的信息量
value[find(i)]+=j;
}
}
for(k=1;k<=n;k++){
if(k==1){
printf("%d",value[find(k)]+dis[k]);
}else{
printf(" %d",value[find(k)]+dis[k]);
}
}
return 0;
}