2435: [Noi2011]道路修建

题目链接

题目大意：n个点，建立生成树，边的费用等于道路长度乘以道路两端的国家个数之差的绝对值，给定方案，求费用

题解：树形dp，会dfs就可以做了

我的收获：这是noi……

#include <iostream>#include <cstdio>#include <algorithm>#include <cstring>using namespace std;const int M=1000005;int n,t,head[M],sz[M];long long ans;struct edge{int to,nex,val;}e[M*2];void add(int i,int j,int w){e[t].to=j,e[t].nex=head[i],e[t].val=w,head[i]=t++;}void dfs(int x,int fa){    sz[x]=1;    for(int i=head[x];i!=-1;i=e[i].nex){        int v=e[i].to;        if(v==fa) continue;        dfs(v,x);sz[x]+=sz[v];        ans+=(long long)e[i].val*abs((n-sz[v])-sz[v]);// have known sz[v],not sz[x]……     }}void work(){    dfs(1,0);    cout<<ans<<endl;}void init(){    int x,y,z;    cin>>n;t=0;    memset(head,-1,sizeof(head));    for(int i=1;i<n;i++)        scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);}int main(){    init();    work();    return 0;}