所謂dsu on tree,就是樹上的啓發式合併,處理出重兒子然後搞就行了。樹的特殊性會使寫起來比數據結構啓發式合併方便一點。只需維護一個數據結構,支持插入刪除單個元素。
這題就是模板題啦…
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=100005,maxe=200005;
int n,a[maxn],sz[maxn],hvy[maxn],fir[maxn],nxt[maxe],son[maxe],tot;
LL ans[maxn],cnt[maxn],sum[maxn],_max;
void add(int x,int y){
son[++tot]=y; nxt[tot]=fir[x]; fir[x]=tot;
}
void dfs_info(int x,int pre){
sz[x]=1;
for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre){
dfs_info(son[j],x); sz[x]+=sz[son[j]];
if(sz[hvy[x]]<sz[son[j]]) hvy[x]=son[j];
}
}
void Push(int x){
sum[cnt[x]]-=x; cnt[x]++; sum[cnt[x]]+=x;
if(cnt[x]>_max) _max=cnt[x];
}
void Del(int x){
sum[cnt[x]]-=x; cnt[x]--; sum[cnt[x]]+=x;
if(!sum[_max]) _max--;
}
void dfs_Push(int x,int pre){
Push(a[x]);
for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre) dfs_Push(son[j],x);
}
void dfs_Del(int x,int pre){
Del(a[x]);
for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre) dfs_Del(son[j],x);
}
void dfs(int x,int pre){
for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre) if(son[j]!=hvy[x]) dfs(son[j],x), dfs_Del(son[j],x);
if(hvy[x]) dfs(hvy[x],x); Push(a[x]);
for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre) if(son[j]!=hvy[x]) dfs_Push(son[j],x);
ans[x]=sum[_max];
}
int main(){
freopen("cf600E.in","r",stdin);
freopen("cf600E.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]), sum[0]+=a[i];
for(int i=1;i<=n-1;i++){
int x,y; scanf("%d%d",&x,&y);
add(x,y); add(y,x);
}
dfs_info(1,1);
dfs(1,1);
for(int i=1;i<=n;i++) printf("%I64d ",ans[i]);
return 0;
}