题意:
有 N 个农民, 他们住在 N 个不同的村子里. 这 N 个村子形成一棵树.
每个农民初始时获得 X 的钱.
每一次操作, 一个农民可以从它自己的钱中, 取出任意数量的钱, 交给某个相邻村子的农民.
对于每个农民给定一个值 v_i, 求
(1) 最少需要多少次操作, 使得每个农民最终拿到的钱 >= 给定的值.
题解:
想到树形背包dp就很好做了。
显然每对人只会交易一次,所以总数不超过n-1次
f
[
i
]
[
j
]
f[i][j]
f[i][j]表示i这个子树,最多交易j次,根节点最多有多少钱(可以为负)
按子树大小合并即可。
code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
const LL inf=1LL<<55;
LL n,X,v[2010];
LL f[2010][2010],g[2010],tot[2010];
struct node{
LL y,next;
}a[4010];LL len=0,last[2010];
void ins(LL x,LL y)
{
a[++len].y=y;
a[len].next=last[x];last[x]=len;
}
void merge(LL x,LL y)
{
for(LL i=0;i<tot[x];i++) g[i]=f[x][i];
for(LL i=0;i<tot[x]+tot[y];i++) f[x][i]=-inf;
for(LL i=0;i<tot[x];i++)
for(LL j=0;j<tot[y];j++)
{
if(f[y][j]>=0) f[x][i+j]=max(f[x][i+j],g[i]);
f[x][i+j+1]=max(f[x][i+j+1],g[i]+f[y][j]);
}
tot[x]+=tot[y];
for(LL i=1;i<tot[x];i++) f[x][i]=max(f[x][i-1],f[x][i]);
}
void dfs(LL x,LL fa)
{
tot[x]=1;f[x][0]=f[x][1]=X-v[x];
for(LL i=last[x];i;i=a[i].next)
{
LL y=a[i].y;
if(y==fa) continue;
dfs(y,x);merge(x,y);
//printf("merge:%lld->%lld\n",y,x);
//printf("x %lld:",x);for(LL i=0;i<tot[x];i++) printf("%lld ",f[x][i]);printf("\n");
}
//printf("x %lld:",x);for(LL i=0;i<tot[x];i++) printf("%lld ",f[x][i]);printf("\n");
}
int main()
{
scanf("%lld %lld",&n,&X);
for(LL i=1;i<=n;i++) scanf("%lld",&v[i]);
for(LL i=1;i<n;i++)
{
LL x,y;scanf("%lld %lld",&x,&y);
ins(x,y);ins(y,x);
}
dfs(1,0);
for(LL i=0;i<n;i++) if(f[1][i]>=0) {printf("%lld",i);return 0;}
}