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; }
不怕路长,只怕心老.