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;
}
上一篇:[Codeforces743D][luogu CF743D]Chloe and pleasant prizes[树状DP入门][毒瘤数据]


下一篇:树状DP (poj 2342)