bzoj1827[Usaco2010 Mar]gather 奶牛大集会 树上DP

。。挺经典的问题了。
在树上找一个点，让其他所有点到达这个点的距离最小。。
一开始并不知道怎么列出dp式。。
然后发现直接贪心就可以了，一开始先算出1的答案，然后一个个往下移动，代价是固定的。
这种想法我自己都想到过这次没想到有点不应该，可能是因为思考时间不够吧。想题有点急躁了，要注意。

#include<cstdio>#include<algorithm>#include<cstring>#include<iostream>#include<vector>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fd(i,a,b) for(int i=a;i>=b;i--)using namespace std;const int N=1e5+5;typedef long long ll;ll ans,c[N],dis[N],size[N];vector<int> e[N],w[N];int n;ll dfs(int x,int fa){    ll ans=dis[x]*c[x];    size[x]=c[x];    for(int i=0;i<e[x].size();i++)    {        int y=e[x][i],v=w[x][i];        if(y==fa)continue;        dis[y]=dis[x]+v;        ans+=dfs(y,x);        size[x]+=size[y];    }    return ans;}void move(int x,int fa){    for(int i=0;i<e[x].size();i++)    {        int y=e[x][i],v=w[x][i];        if(y==fa)continue;        if(size[1]-2*size[y]<0)        {            ans=ans+(size[1]-2*size[y])*v;            move(y,x);        }    }}int main(){    scanf("%d",&n);    fo(i,1,n)scanf("%lld",&c[i]);    fo(i,1,n-1)    {        int u,v,x;        scanf("%d%d%d",&u,&v,&x);        e[u].push_back(v);        e[v].push_back(u);        w[u].push_back(x);        w[v].push_back(x);    }    ans=dfs(1,0);    move(1,0);    printf("%lld\n",ans);    return 0;}