【bzoj 2435】道路修建（DFS）

最新推荐文章于 2019-10-23 15:50:43 发布

zP1nG

最新推荐文章于 2019-10-23 15:50:43 发布

阅读量404

点赞数

CC 4.0 BY-SA版权

分类专栏： zP1nG的bzoj

本文链接：https://blog.youkuaiyun.com/zP1nG/article/details/79047578

zP1nG的bzoj 专栏收录该内容

183 篇文章

订阅专栏

本文通过一个具体的树形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;
}