浅谈带权并查集

 本文主要从0基础开始讲解带权并查集，要求读者阅读本文前对并查集有一定的了解。
 还是先以一道木板题为例吧！

模板(hdu 3038)

大致题意：

  给出区间[a, b]，区间之和为v。输入m组数据，每输入一组，判断此组条件是否与前面冲突，最后输出与前面冲突的数据的个数。
比如说先给出[1,10]的和为100,[1,3]的和为10，[7,10]的和为20，接着又说[4,6]的和为50，显然此时发生冲突。

分析：

  读者多试几个样例不难看出其实这道题就是想让你统计区间的端点，如果出现例如上面所说的“端点重现”的情况（但要要求端点要被覆盖过），就有可能发生冲突。
于是我们就可以用并查集（显然需要带权）来统计端点出现情况。其实就是用区间建立并查集。

下面我结合代码来具体讲解这个带权并查集。

#include<bits/stdc++.h>
using namespace std;

const int MAX=2e5+20;

int n,m,ans;
int father[MAX],d[MAX];//d数组表示每个元素到他的父节点这条边的权值

template<typename T> void Read(T &cn)//读入优化，这道题其实不用
{
	char c; int sig=1;
	while(!isdigit(c=getchar())) if(c=='-') sig=-1; cn=c-48;
	while(isdigit(c=getchar())) cn=cn*10+c-48; cn*=sig;
}

template<typename T> void Write(T cn)//输入优化，这道题其实不用
{
	if(cn<0) {putchar('-'); cn=0-cn;}
	T cm=0; int wei=0; int cx=cn%10; cn/=10;
	while(cn) cm=cm*10+cn%10,cn/=10,wei++;
	while(wei--) putchar(cm%10+48),cm/=10;
	putchar(cx+48);
}

void INIT()
{
	ans=0;
	for(int i=1;i<=n+1;i++) father[i]=i,d[i]=0;
}

int find(int x)
{
	int t=father[x];
	if(x!=father[x]) father[x]=find(father[x]);
	d[x]+=d[t];
	return father[x];
}

void merge(int a,int b,int fa,int fb,int s)
{
	father[fb]=fa;//我这儿是让小元素作为大元素祖先
	d[fb]=d[a]+s-d[b];//我在代码下面有张图，可以辅助理解，应该不难推出该结论
}

void solve(int a,int b,int s)
{
	int fa=find(a);//找祖先
	int fb=find(b);
	if(fa!=fb) merge(a,b,fa,fb,s);//需要合并区间端点集
	else
	{
		if(d[b]-d[a]!=s) ans++;//发生冲突
	}
}

int main()
{
	freopen("1.in","r",stdin);
	while(scanf("%d %d",&n,&m)!=EOF)//注意这道题其实有多组数据（题目没说，但数据是）
	{
		INIT();
		while(m--)
		{
			int a,b,s;
			Read(a); Read(b); Read(s); b++;
			//这里的b++，这样才能真正覆盖整个区间，否则下次出现冲突情况时，并查集中却没有相同的两个端点
			//也可以a--，必须有一边要多扩展一个单位
			//这里请读者好好想想，可以结合样例分析一下
			solve(a,b,s);
		}
		Write(ans); printf("\n");
	}
	return 0;
}

图：

好了，这就是传说中的带权并查集了。
读者可以再看看这道题poj 1182。

学习厌倦了？点我有更多精彩哦！