題目鏈接

題意：給你一棵樹，各個節點都有價值（除根節點），從根節點出發，選擇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;
}

轉載于:https://www.cnblogs.com/neng18/p/3676391.html

總結

以上是生活随笔為你收集整理的hdu 1561 The more, The Better_树状dp的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。