hdu 1561 The more, The Better_树状dp

题目链接

题意：给你一棵树，各个节点都有价值（除根节点），从根节点出发，选择m个节点，问最多的价值是多小。

思路：很明显是树状dp，遍历树时背包最优价值，dp[i][k]=max{dp[i][r]+dp[son[i]][k-r]}

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std; 
#define MAXN 250
struct node{
	int from,to,next;
}edge[MAXN];
int tot,head[MAXN],visit[MAXN],value[MAXN],dp[MAXN][MAXN],f[MAXN][MAXN];
int n,m;
void add(int a,int b){//链式向前星 
	edge[tot].from=a;
	edge[tot].to=b;
	edge[tot].next=head[a];
	head[a]=tot++;
}
void dfs(int root){
	int i,j,u,k;
	visit[root]=1;
	for(i=head[root];i!=-1;i=edge[i].next){
		u=edge[i].to;
		if(!visit[u]){
			dfs(u);//访问下一个节点 
			for(k=m;k>=0;k--){//可以选择m个节点 
				for(j=0;j<=k;j++){
					f[root][k]=max(f[root][k],f[root][k-j]+dp[u][j]);//背包子节点的价值
				}
			}
		}
	}
	for(j=1;j<=m+1;j++)
		dp[root][j]=f[root][j-1]+value[root];//加上自己节点的价值
}
int main(int argc, char** argv) {
	int i,a,b;
	while(scanf("%d%d",&n,&m)!=EOF,n+m){
		tot=0;
		memset(head,-1,sizeof(head));
		for(i=1;i<=n;i++){//从节点1开始，保留节点0为根节点
			scanf("%d%d",&a,&b);
			add(a,i);//单向 
			value[i]=b;
		}
		value[0]=0;//0为根节点，价值为零
		memset(visit,0,sizeof(visit));
		memset(dp,0,sizeof(dp));
		memset(f,0,sizeof(f));
		dfs(0);//从0开始遍历树
		printf("%d\n",dp[0][m+1]);
	} 
	return 0;
}