3090: Coci2009 [podjela]

题意：

有 N 个农民, 他们住在 N 个不同的村子里. 这 N 个村子形成一棵树.
每个农民初始时获得 X 的钱.
每一次操作, 一个农民可以从它自己的钱中, 取出任意数量的钱, 交给某个相邻村子的农民.

对于每个农民给定一个值 v_i, 求
(1) 最少需要多少次操作, 使得每个农民最终拿到的钱 >= 给定的值.

题解：

想到树形背包dp就很好做了。
显然每对人只会交易一次，所以总数不超过n-1次
$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;}
}