[dsu on tree] Codeforces #600E. Lomsat gelral

本文详细介绍了一种称为DSU on Tree的算法实现过程,该算法适用于树状结构的数据处理,通过启发式合并来优化计算效率。文章提供了一个具体的应用案例,并附带完整的代码示例。

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

所谓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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值