JZOJ4814. 【NOIP2016提高A组五校联考2】tree

Description

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

Input

第一行为两个整数n; lim
接下来n 行，第i 行包含一个整数vi，表示结点i 上物品的价值。
接下来n- 1 行，每行包含两个整数u; v, 描述一条连接u; v 结点的树边。

Output

输出一行答案。

Sample Input

6 4
-5
4
-6
6
9
6
3 2
3 1
2 4
2 5
1 6

Sample Output

2

Data Constraint

对于前20% 的数据，1<=n; lim<=10
对于前60% 的数据，1<=n; lim<=100
对于100% 的数据，1<=n; lim<=3000; |vi| <=10^5 数据有梯度，保证给出的是合法的树。

分析

这题很显然，就是要用树形DP。
但是，怎样将子树的状态转移到树的节点？

我们已经知道了每一棵子树选x个的最大值，
我们就将它认为是一个重量为x，价值为最大值的物品。
这里就可以用背包了。

但是，在每一棵子树上面都只能选一个物品，
所以背包就需要一些小修改。

code(c++)

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h> 
using namespace std;
int next[6003],a[6003],b[3003],v[3003],f[3003][3003],n,m,x,y,ans,g[3003];
void dg(int x,int fa,int deep)
{
    bool pd=1;
    g[x]=1;
    for(int i=b[x];i;i=next[i])
        if(fa!=a[i])dg(a[i],x,deep+1),pd=0,g[x]+=g[a[i]];
    g[x]=min(g[x],m-deep);
    if(pd)
    {
        f[x][0]=0;
        f[x][1]=v[x];
        g[x]=1;
        for(int j=2;j<=m;j++)
            f[x][j]=-10000000;
        return;
    }
    f[x][0]=0;
    f[x][1]=v[x];
    for(int j=1;j<=g[x];j++)
        f[x][j]=v[x];
    for(int i=b[x];i;i=next[i])
        if(a[i]!=fa)
        {
            for(int j=g[x];j>1;j--)
            {
                for(int k=1;k<min(j,g[a[i]]+1);k++)
                    f[x][j]=max(f[x][j],f[x][j-k]+f[a[i]][k]);
            }
        }
}
int main()
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&v[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        next[i]=b[x];
        a[i]=y;
        b[x]=i;
        next[i+n]=b[y];
        a[i+n]=x;
        b[y]=n+i;
    }
    dg(1,0,0);
    ans=-2147483647;
    for(int i=m;i;i--)
        ans=max(ans,f[1][i]);
    printf("%d",ans);
}