[NOI2011]道路修建[搜索]

P2052 [NOI2011]道路修建

好水.... 我居然十分钟就切了它QAQ

先任意选一个点为根计算出它各子树大小 然后再遍历 来算总花费

注意开long long！！！

#include<bits/stdc++.h>
using namespace std;
#define Max(x,y) (x)>(y)?(x):(y)
#define Min(x,y) (x)>(y)?(y):(x)
#define ll long long
#define rg register
const int N=1000000+5,M=10000+5,inf=0x3f3f3f3f,P=99999997;
map<string,int>a;
int n,m;
ll size[N];
ll ans=0;
bool vis[N];
template <class t>void rd(t &x){
    x=0;int w=0;char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
    x=w?-x:x;
}

int head[N<<1],tot=0;
struct edge{int v,w,nxt;}e[N<<1];
void add(int u,int v,int w){
    e[++tot]=(edge){v,w,head[u]},head[u]=tot;
}

void dfs(int u){
    vis[u]=1,size[u]=1;
    for(int i=head[u],v;i;i=e[i].nxt){
        v=e[i].v;
        if(vis[v]) continue;
        dfs(v);
        size[u]+=size[v];
    }
}

void dfs2(int u){
    vis[u]=1; 
    for(int i=head[u],v;i;i=e[i].nxt){
        v=e[i].v;
        if(vis[v]) continue;
        dfs2(v);
        ans+=(ll)abs(size[1]-(size[v]<<1))*e[i].w;
    }
}

int main(){
    freopen("in.txt","r",stdin);
    rd(n);
    for(int i=1,u,v,w;i<n;++i) rd(u),rd(v),rd(w),add(u,v,w),add(v,u,w);
    dfs(1);
    memset(vis,0,sizeof(vis));
    dfs2(1);
    printf("%lld",ans);
    return 0;
}