题目
给出以1为根的一棵树,树的每个节点上都有编号,求每个节点所在子树中最多的数字是哪个?(有相同数目的,就把编号相加输出)
题解
dsu on tree的模板题
先用树链剖分的方法,标记出重儿子,然后遍历一遍整棵树,将重儿子的编号数组保留,轻链上的儿子的编号数组用完后就清楚,每个节点的结果为遍历轻儿子后的结果直接加上重儿子数组后的结果,如果不保留重儿子的数组,那么每个点的结果需要遍历他的所有儿子才能知道,肯定会T
关于为什么删除轻儿子的编号数组
一开始看我确实挺疑惑的,既然保留数组能够减少时间复杂度,那为什么不能都保留呢,再看看别人的博客,感觉大概明白 (其实并没有),就是对一个节点遍历所有子树时,如果保留上一个子树的数组,那么会影响到下一个下一个子树的数组的计算,所以想要保留数组的话,肯定得最后一个计算,然后就是各个的子树到底应该哪个最后一个计算呢,当时是重链了
要是我讲的不够明白(还是太菜了),强烈推荐这篇博客,讲的是真的清楚
树上启发式合并
完整代码
#include<iostream>
using namespace std;
#define ll __int64
const int maxn=1e5+5;
int vis[maxn],son[maxn<<1],head[maxn],fa[maxn],sz[maxn],h[maxn],a[maxn],cnt,Max;
ll c,ans[maxn];
struct E
{
int to,nxt;
}edge[maxn<<1];
void add_edge(int x,int y)
{
edge[++cnt].nxt=head[x];
edge[cnt].to=y;
head[x]=cnt;
}
void dfs(int x,int f)
{
fa[x]=f;sz[x]=1;
for(int i=head[x];i;i=edge[i].nxt){
int v=edge[i].to;
if(v!=f){
dfs(v,x);
sz[x]+=sz[v];
if(sz[son[x]]<sz[v]) son[x]=v;
}
}
}
void cal(int x,int p)
{
//cout<<x<<' '<<p<<endl;
h[a[x]]+=p;
if(p>0&&h[a[x]]==Max) c+=a[x];
if(p>0&&h[a[x]]>Max) Max=h[a[x]],c=a[x];
for(int i=head[x];i;i=edge[i].nxt){
int v=edge[i].to;
if(v!=son[x]&&v!=fa[x]) cal(v,p);
}
}
void Dfs(int x,int p)
{
//cout<<x<<' '<<p<<endl;
for(int i=head[x];i;i=edge[i].nxt){
int v=edge[i].to;
if(v!=fa[x]&&v!=son[x]) Dfs(v,0);
}
if(son[x]) Dfs(son[x],1),vis[son[x]]=1;
cal(x,1);ans[x]=c;
if(son[x]) vis[son[x]]=0;
if(!p) cal(x,-1),c=Max=0;
}
int main()
{
int n,x,y;
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n-1;i++){
scanf("%d%d",&x,&y);
add_edge(x,y);
add_edge(y,x);
}
dfs(1,0);Dfs(1,0);
for(int i=1;i<=n;i++){
cout<<ans[i];
if(i!=n) cout<<' ';
}
cout<<endl;
}