【bzoj2435】 NOI2011 道路修建 水题

其实不能说什么了，貌似你会dfs这道题就能A，当然考场上要写bfs。。。

直接放代码了。。。

#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<iostream>#include<algorithm>#define maxn 1000010using namespace std;int n,m,num;long long ans;int next[2*maxn],to[2*maxn],head[maxn];long long len[2*maxn];int size[maxn];void addedge(int x,int y,int z){num++;to[num]=y;len[num]=z;next[num]=head[x];head[x]=num;}void dfs(int x,int fa){size[x]=1;for (int p=head[x];p;p=next[p])  if (to[p]!=fa)  {  dfs(to[p],x);  size[x]+=size[to[p]];  ans+=(long long)len[p]*abs(n-size[to[p]]-size[to[p]]);  }}int main(){scanf("%d",&n);for (int i=1;i<n;i++){int x,y,z;scanf("%d%d%d",&x,&y,&z);addedge(x,y,z);addedge(y,x,z);}dfs(1,0);printf("%lld\n",ans);return 0;}