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;
}