树上启发式合并模板（更好的暴力）

xqx_Zi_yu

于 2020-11-10 11:26:55 发布

阅读量161

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： dfs 树上启发式合并

本文链接：https://blog.csdn.net/qq_44132777/article/details/109595484

算法专栏收录该内容

59 篇文章

订阅专栏

本文介绍了一种基于轻重链剖分算法的实现方法，该算法用于解决树形结构中路径更新和查询的问题。通过递归遍历树结构来确定每个节点的轻重儿子，并利用递归加线性更新的方式计算每条路径的最大值和总和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

以CF600E Lomsat gelral为例

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
typedef long long ll;
int a[maxn],sz[maxn],son[maxn];
int n;
vector<int> G[maxn];
//遍历确定轻重儿子 
void dfs(int u,int fa){
	sz[u]=1;
	for(int i=0;i<G[u].size();++i){
		int v=G[u][i];
		if(v==fa) continue;
		dfs(v,u);
		sz[u]+=sz[v];
		if(sz[v]>sz[son[u]]) son[u]=v;
	}
}
ll cnt[maxn],sum,maxx,Son,ans[maxn];
void add(int u,int fa,int val){		//val=1加上贡献 val=-1消除贡献 
	cnt[a[u]]+=val;
	if(cnt[a[u]]>maxx) maxx=cnt[a[u]],sum=a[u];
	else if(cnt[a[u]]==maxx) sum+=a[u];
	for(int i=0;i<G[u].size();++i){
		int v=G[u][i];
		if(v==fa||v==Son) continue;    
		add(v,u,val);
	}
}
void dfs2(int u,int fa,int xp) {	//xp=1保存 xp=0 消除 
	for(int i=0;i<G[u].size();++i){
		int v=G[u][i];
		if(v==fa||v==son[u]) continue;	//先遍历轻儿子
		dfs2(v,u,0);
	}
	if(son[u]) dfs2(son[u],u,1),Son=son[u];	//遍历重儿子 
	add(u,fa,1); Son=0;
	ans[u]=sum;
	if(xp==0) {
		add(u,fa,-1);sum=0,maxx=0;
	}
}
signed main() {
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",a+i);
	for(int i=1;i<n;i++) {
		int u,v; scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1,0);
	dfs2(1,0,1);
	for(int i=1;i<=n;i++) 
		printf("%lld ",ans[i]);
}