算法之并查集和带权并查集

并查集
一、解决的问题
并查集被很多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;
	
} 

3、参考资料
带权并查集
网络分析