P2986 Great Cow Gathering G

 

换根dp, father -> son  , 基本是加减

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50

#include <bits/stdc++.h> using namespace std ;  const int N=1e5+2,M=N*5;  #define int long long  int n,a[N],sz[N],g[N],f[N],S;  int nxt[M],go[M],w[M],hd[N],all;     void add(int x,int y,int z){     go[++all]=y,nxt[all]=hd[x],w[all]=z;     hd[x]=all;  }  void d0(int x,int fa){     int i,y,z;     sz[x]=a[x];             for(i=hd[x];i;i=nxt[i]){          y=go[i],z=w[i];          if(y!=fa) d0(y,x),sz[x]+=sz[y];       }       for(i=hd[x];i;i=nxt[i]){         y=go[i],z=w[i];         if(y!=fa) g[x]+=g[y]+sz[y]*z;       }  }     void dfs(int x,int fa){     int i,y,z;             for(i=hd[x];i;i=nxt[i]){          y=go[i],z=w[i];          if(y!=fa){             f[y]=f[x]-z*sz[y]+z*(S-sz[y]);             dfs(y,x);          }       }  }  signed main(){     int i,x,y,z,t=1e9;     cin>>n;     for(i=1;i<=n;i++) cin>>a[i],S+=a[i];     for(i=1;i<n;i++) cin>>x>>y>>z,add(x,y,z),add(y,x,z);     d0(1,0);     dfs(1,0);           for(i=1;i<=n;i++) t=min(t,f[i]);           cout<<t+g[1]<<'\n';  }