【jzoj 4814】【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 数据有梯度，保证给出的是合法的树。

The Solution

很容易想到树形dp。

我们可以设f[i][j]表示以i为根的子树中，选了j个节点，所获得的最大价值。

然后随便推推就可以推出来了，题目可以转换成一个背包问题。
可以优化的是，你枚举的时候，可以以min(size[x],lim)为上下限枚举，size[x]表示你以x为根的子树的总节点数。

//DFS(x,y) 表示当前枚举第x个节点的根节点是y

CODE

#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstring>
#define fo(i,a,b) for (int i=a;i<=b;i++)
#define fd(i,a,b) for (int i=a;i>=b;i--)
#define N 3005
#define INF 2147483647

using namespace std;

int n,lim,x,y,tot=0,ans=0;
int val[N],size[N],final[N * 2],f[N][N];

struct node
{
    int to,next;

    node(void){}
    node(int a,int b) : to(a),next(b){}
}e[N * 2];

void link(int u,int v)
{
    e[++ tot] = node(v,final[u]),final[u] = tot;
    e[++ tot] = node(u,final[v]),final[v] = tot;
}

void Dfs(int x,int y)
{
    f[x][0] = 0;
    f[x][1] = val[x];
    size[x]++;
    for (int k = final[x];k;k = e[k].next)
    {
        if (e[k].to == y) continue;
        Dfs(e[k].to,x);
        size[x] += size[e[k].to];
        int mn=min(size[x],lim);
        fd(i,mn,0)
        {
            int q=min(size[e[k].to],i-1);
            fo(j,1,q) f[x][i] = max(f[x][i],f[x][i-j] + f[e[k].to][j]);
        }
    }
}

int main() 
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    scanf("%d%d",&n,&lim);
    fo(i,1,n) scanf("%d",&val[i]);
    fo(i,1,n-1) 
    {
        scanf("%d%d",&x,&y),
        link(x,y);
    }
    memset(f,128,sizeof(f));
    Dfs(1,0);
    fo(i,1,lim) ans=max(ans,f[1][i]);
    printf("%d\n",ans);
    return 0;
}