BZOJ2435: [Noi2011]道路修建

最新推荐文章于 2018-10-09 14:50:00 发布

DYT_B

最新推荐文章于 2018-10-09 14:50:00 发布

阅读量213

点赞数

CC 4.0 BY-SA版权

分类专栏：题解 BZOJ题解

本文链接：https://blog.youkuaiyun.com/DYT_B/article/details/80098749

题解同时被 2 个专栏收录

125 篇文章

订阅专栏

BZOJ题解

67 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）进行节点计数的方法，并通过实例代码展示了如何实现该算法来解决特定问题。该算法应用于图遍历场景中，通过对每个节点的递归访问计算子树大小并统计特定条件下的路径数量。

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

题目描述：

传送门

题解：

dfs计数？

#include<cstdio>
#include<string>
using namespace std;
const int maxn=1000005;
int n,tot,lnk[maxn],son[2*maxn],nxt[2*maxn],w[2*maxn];
bool vis[maxn];
long long ans;
inline int read(){
    int x=0; char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x;
}
void add(int x,int y,int z){
    son[++tot]=y,w[tot]=z,nxt[tot]=lnk[x],lnk[x]=tot;
}
int abs_(int x) {if (x<0) return -x; else return x;}
int dfs(int x){
    int sum=0;
    for (int j=lnk[x];j;j=nxt[j])
    if (!vis[son[j]]){
        vis[son[j]]=1; int siz=dfs(son[j]); sum+=siz;
        ans+=1ll*abs_(n-siz-siz)*w[j];
    }
    return sum+1;
}
int main(){
    n=read();
    for (int i=1;i<n;i++) {
        int x=read(),y=read(),z=read();
        add(x,y,z); add(y,x,z);
    }
    vis[1]=1; dfs(1);
    printf("%lld\n",ans);
    return 0;
}