【JZOJ 4814】【NOIP2016提高组 五校联考2】tree

Description

给一棵n 个结点的有根树，结点由1 到n 标号，根结点的标号为1。每个结点上有一个物品，第i 个结点上的物品价值为vi。
你需要从所有结点中选出若干个结点，使得对于任意一个被选中的结点，其到根的路径上所有的点都被选中，并且选中结点的个数不能超过给定的上限lim。在此前提下，你需要最大化选中结点上物品的价值之和。
求这个最大的价值之和。
对于100% 的数据，1<=n; lim<=3000; |vi| <=10^5 数据有梯度，保证给出的是合法的树。

Analysis

直接树形DP就好了
F[v][x]表示v及其子树选了x个点满足条件的最大值。
一个一个儿子分别与v合并。
看似三重循环，好像要n*lim^2
但是如果子树多那么每个子树的大小就小，相互制约。
于是好像以难以分析的时间复杂度O（跑得很快）过掉了

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
#define efo(i,v) for(int i=last[v];i;i=next[i])
using namespace std;
const int N=3010,M=N*2;
int n,lim,tot,a[N],to[M],next[M],last[N],f[N][N];
void link(int u,int v)
{
    to[++tot]=v,next[tot]=last[u],last[u]=tot;
}
int dfs(int v,int fr)
{
    f[v][1]=a[v];
    int size=1;
    efo(i,v)
    {
        int u=to[i];
        if(u==fr) continue;
        int s=dfs(u,v);
        fd(x,size,1)
            fd(y,min(s,lim-x),1) f[v][x+y]=max(f[v][x+y],f[v][x]+f[u][y]);
        size+=s;
    }
    return size;
}
int main()
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    int u,v,ans=0;
    scanf("%d %d",&n,&lim);
    fo(i,1,n) scanf("%d",&a[i]);
    fo(i,1,n-1)
    {
        scanf("%d %d",&u,&v);
        link(u,v),link(v,u);
    }
    memset(f,128,sizeof(f));
    dfs(1,1);
    fo(i,1,lim) ans=max(ans,f[1][i]);
    printf("%d",ans);
    return 0;
}