【bzoj 2435】道路修建(DFS)

本文通过一个具体的树形DP题目,展示了如何避免系统栈崩溃的问题,并分享了一个高效的实现方法。采用递归的方式,计算树中每条路径的价值,最终求得整棵树的总价值。

传送门biu~
第一眼看:这不是sb题么?
实际上当树的深度特别大的时候,递归的层数过深,系统栈就特别容易崩溃。之前在dfs的时候传了两个参数,一直RE。还好有Duan2baka大佬指教,改成一个参数才可以Ac掉这道题。

#include<bits/stdc++.h>
using namespace std;
int n;
int head[1000005],siz[1000005],nex[2000005],to[2000005],val[2000005],tp=0;
long long ans=0;
inline void add(int x,int y,int z){
    nex[++tp]=head[x];
    head[x]=tp;
    to[tp]=y;
    val[tp]=z;
}
void dfs(int x){
    siz[x]=1;
    for(int i=head[x];i;i=nex[i]){
        if(siz[to[i]])  continue;
        dfs(to[i]);
        ans+=1ll*val[i]*abs(n-siz[to[i]]-siz[to[i]]);
        siz[x]+=siz[to[i]];
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;++i){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);
    }
    dfs(1);
    printf("%lld",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

zP1nG

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值