第十一届蓝桥杯 网络分析 带权并查集

思路:要么合并两个集合,要么给集合所有结点加数字
d数组表示当前结点和根节点的差,b数组表示根结点所在集合里面所有数字要增加的数字
注意合并的时候,作为儿子结点的点的d数组要更新为原来两个集合之前的差值
最后每个数字答案就是块的值加上自身与块所在集合的差值

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+50;
int f[N],d[N],b[N];
int find(int x){
    
    
	if(f[x]!=x){
    
    
		int fa=f[x];
		f[x]=find(f[x]);
		d[x]+=d[fa];
	}
	return f[x];
}
int main(){
    
    
	int n,m;cin>>n>>m;
	for(int i=1;i<=n;i++) f[i]=i;
	while(m--){
    
    
		int op,x,y;cin>>op>>x>>y;
		if(op==1){
    
    
			int fx=find(x),fy=find(y);
			if(fx!=fy){
    
    
				f[fx]=fy;
				d[fx]=b[fx]-b[fy];
			}
		}
		else b[find(x)]+=y;
	}
	for(int i=1;i<=n;i++) cout<<b[find(i)]+d[i]<<" ";
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_43563669/article/details/109988606