题目链接
题意:给你一棵树,各个节点都有价值(除根节点),从根节点出发,选择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;
}