hdu1561 (树形DP_01背包)

来源：互联网发布：体重测试软件编辑：程序博客网时间：2024/05/16 11:43

题意：有N个城堡，在其中选M个并拿走其中的物品，看最多能拿到多少个

其中，城堡里的物品数不相同，要拿有些城堡的东西得先拿走指定的另一个城堡的物品。

思路：用dfs+01背包

先是建立一棵树，由于输入的关系是森林，则把这些森林中树的根全指向0，用0来做为这一个大树的根。

其次是要对每个结点的状态进行01背包计算，dp[i][j]表示以结点i为根点向其子孙连续地选j个结点的最大值（其中包括结点本身）

#include<cstdio>#include<iostream>#include<cstring>#include<queue>#include<string>#include<vector>#include<cmath>#include<algorithm>#include<map>using namespace std;const int N=202;int n;int dp[N][N],val[N];vector<int> v[N];void dfs(int d,int m){if (m<=0) return;for (int i=0;i<v[d].size();i++)  //有多少个分去 {int cur=v[d][i];dfs(cur,m-1);for (int j=m;j>=2;j--)   //对单个分去的情况进行DP  ,,这里注意j只能从大到小，从小到大就是多重背包了。 for (int k=1;k<j;k++) dp[d][j]=max(dp[d][j],dp[d][j-k]+dp[cur][k]);}}int main(){freopen("in","r",stdin);int m;while (cin>>n>>m,n){for (int i=0;i<=n;i++) v[i].clear();for (int i=2;i<=m+1;i++) dp[0][i]=0;for (int i=1;i<=n;i++){int a,b;scanf("%d%d",&a,&b);val[i]=b;dp[i][1]=val[i];if (a) v[a].push_back(i);else v[0].push_back(i);for (int j=2;j<=m;j++) dp[i][j]=0; }dfs(0,m+1);cout<<dp[0][m+1]<<endl;}}

0 0