Luogu-T91116 (启发式合并)

题目

题意：给一个树，每个边有个权值。一条路径的代价为路径上的边权中的最小值。求一个点，使得其它点到它的路径代价和最大。

分析

按边权给边排序，然后从大到小依次加入图中，那么每次这个边所连接的两团，相互到达的路径代价就是这个边的权值。用并查集，然后又需要每次对并查集内的所有点加某个数，于是想到可以只给这个集合的代表点加，然后用启发式合并保证时间复杂度。最后按新的有根扫一遍，把每个点的答案（即选这个点的话路径代价和），取最大值即可。

关于启发式合并，就是说并查集在查询的时候，不进行路径压缩，在合并的时候，总是将点个数少的集合合并到点个数多的集合。这样就保证了每个点合并次数不超过 lgn，使得最终形成的树深度不超过 lgn。

时间复杂度是 O(n lgn) （大概吧）

代码

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

int n;
int fa[MAXN];
int cnt[MAXN];
long long S[MAXN];


vector<pair<long long, int>> graph[MAXN];
vector<pair<long long, pair<int, int>>> E;

vector<int> gg[MAXN];



int Fa(int X) {return fa[X]==X ? X : Fa(fa[X]);}

void ADD(int U, int V, long long W)
{
	int fu,fv;
	fu=Fa(U);
	fv=Fa(V);
	
	if (cnt[fu]>cnt[fv]) swap(fu,fv);	//启发式合并
	
	S[fv]+=W*cnt[fu];
	S[fu]+=W*cnt[fv]-S[fv];
	
	cnt[fv]+=cnt[fu];
	cnt[fu]=0;
	fa[fu]=fv;
	gg[fv].push_back(fu);
}


void dfs(int X, int fX)
{
	S[X]+=S[fX];
	for (auto o: gg[X]) if (o!=fX)
		dfs(o,X);
}


void solve()
{
	scanf("%d",&n);
	for (int i=1; i<=n; i++) {
		fa[i]=i;
		cnt[i]=1;
	}
	for (int i=1,uu,vv,ww; i<n; i++) {
		scanf("%d%d%d",&uu,&vv,&ww);
		graph[uu].push_back(make_pair(ww,vv));
		graph[vv].push_back(make_pair(ww,uu));
		E.push_back(make_pair(-ww,make_pair(uu,vv)));
	}
	sort(E.begin(),E.end());
	
	for (auto ee: E) {
		ADD(ee.second.first, ee.second.second, -ee.first);
	}
	
	int root;
	for (int i=1; i<=n; i++)
		if (Fa(i)==i) root=i;
	
	dfs(root,0);
	
	long long ans=-1;
	for (int i=1; i<=n; i++) ans=max(ans, S[i]);
	
	printf("%lld\n",ans);
}

int main()
{
	solve();
	return 0;
}