BZOJ2435: [Noi2011]道路修建

题目描述:

传送门

题解:

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;
}
    原文作者:道路修建问题
    原文地址: https://blog.csdn.net/DYT_B/article/details/80098749
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞