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;
}