CF600E Lomsat gelral
II.I.CF600E Lomsat gelral
dot 的流程如下:
- 遍历轻儿子,更新轻儿子子树内答案。在更新完后,删除轻儿子子树内影响(即清空数组)。
- 遍历重儿子,更新重儿子子树内答案,但不删除。
- 将自身影响及轻儿子影响加入。
- 更新自身答案。
本题就照着这么做就行了。跑得还是比暴力线段树合并要快那么一丢丢的
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,buc[100100],col[100100],mx,son[100100],sz[100100];
ll sum[100100],res[100100];
vector<int>v[100100];
void dfs1(int x,int fa){
sz[x]=1;
for(auto y:v[x])if(y!=fa){
dfs1(y,x),sz[x]+=sz[y];
if(sz[y]>sz[son[x]])son[x]=y;
}
}
void Push(int x){sum[buc[x]]-=x,buc[x]++,sum[buc[x]]+=x;if(mx<buc[x])mx=buc[x];}
void Pop(int x){sum[buc[x]]-=x,buc[x]--,sum[buc[x]]+=x;if(!sum[mx])mx--;}
void dfs2(int x,int fa){//only push bucket
Push(col[x]);
for(auto y:v[x])if(y!=fa)dfs2(y,x);
}
void dfs3(int x,int fa){//only pop bucket
Pop(col[x]);
for(auto y:v[x])if(y!=fa)dfs3(y,x);
}
void dfs4(int x,int fa){//push bucket and get answer
for(auto y:v[x])if(y!=fa&&y!=son[x])dfs4(y,x),dfs3(y,x);
if(son[x])dfs4(son[x],x);
Push(col[x]);
for(auto y:v[x])if(y!=fa&&y!=son[x])dfs2(y,x);
res[x]=sum[mx];
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&col[i]);
for(int i=1,x,y;i<n;i++)scanf("%d%d",&x,&y),v[x].push_back(y),v[y].push_back(x);
dfs1(1,0);
dfs4(1,0);
for(int i=1;i<=n;i++)printf("%lld ",res[i]);
return 0;
}