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
,dis
,size
;
vector<int> e
,w
;
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;
}