【DFS】bzoj2435 [Noi2011]道路修建

两遍DFS。第一遍统计以每个点为根的子树大小，第二遍更新答案。

 #include<cstdio>
 #include<iostream>
 using namespace std;
 int v[],w[],first[],next[],en,sz[];
 bool vis[];
 long long ans;
 inline int Abs(const int &x){return x< ? -x : x;}
 inline void AddEdge(const int &U,const int &V,const int &W)
 {v[++en]=V;w[en]=W;next[en]=first[U];first[U]=en;}
 int n,a,b,c,res;
 char C;
 inline int Get()
 {
     res=;C='*';
     while(C<''||C>'')C=getchar();
     while(C>=''&&C<=''){res=res*+(C-'');C=getchar();}
     return res;
 }
 void dfs(int cur)
 {
     vis[cur]=true;
     sz[cur]=;
     for(int i=first[cur];i;i=next[i])
       if(!vis[v[i]]){dfs(v[i]);sz[cur]+=sz[v[i]];}
     vis[cur]=false;
 }
 void dfs2(int cur)
 {
     vis[cur]=true;
     for(int i=first[cur];i;i=next[i])
       if(!vis[v[i]])
         {
           ans+=(long long)Abs((sz[v[i]]<<)-n)*w[i];
           dfs2(v[i]);
         }
     vis[cur]=false;
 }
 int main()
 {
     n=Get();
     for(int i=;i<n;i++){a=Get();b=Get();c=Get();AddEdge(a,b,c);AddEdge(b,a,c);}
     dfs();
     dfs2();
     cout<<ans<<endl;
     return ;
 }