【JZOJ4814】tree

Description

给一棵n 个结点的有根树，结点由1 到n 标号，根结点的标号为1。每个结点上有一个物品，第i 个结点上的物品价值为vi。
你需要从所有结点中选出若干个结点，使得对于任意一个被选中的结点，其到根的路径上所有的点都被选中，并且选中结点的个数不能超过给定的上限lim。在此前提下，你需要最大化选中结点上物品的价值之和。
求这个最大的价值之和。

Solution

树形dp，设$F_{i,j}$表示以$i$为根的子树选了$j$个节点的最大值。

那么我们转移要枚举儿子选的个数，这样会超时。

我们考虑换一种想法，枚举已经选择过的，与还没有选择的个数，那么我们转移只要$n^2$，那么这相当于一个背包问题，第二维是空间，价值就是 F <script type="math/tex" id="MathJax-Element-522">F</script>值。

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define N 3010
#define M 6020
using namespace std;
int n,lim;
int f[N][N];
int to[M],next[M],last[M],num=0;
int v[N],siz[N],fa[N];
void link(int x,int y)
{
    num++;
    to[num]=y;
    next[num]=last[x];
    last[x]=num;
}
void dfs(int x)
{
    siz[x]=1;
    f[x][1]=v[x];
    for(int i=last[x];i;i=next[i])
    {
        int v=to[i];
        if(v!=fa[x])
        {
            fa[v]=x;
            dfs(v);
            fd(j,siz[x],1)
            fd(k,siz[v],1)
            f[x][j+k]=max(f[x][j+k],f[x][j]+f[v][k]);
            siz[x]+=siz[v];
        }
    }
}
int main()
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    memset(f,180,sizeof(f));
    cin>>n>>lim;
    fo(i,1,n) scanf("%d",&v[i]);
    fo(i,1,n-1)
    {
        int x,y;
        scanf("%d %d",&x,&y);
        link(x,y);
        link(y,x);
    }
    dfs(1);
    int ans=0;
    fo(i,1,lim) ans=max(ans,f[1][i]);
    cout<<ans;
}