您的位置:首页 > 其它

hdu1561 (树形DP_01背包)

2014-10-22 20:30 169 查看
题意:有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

,val
;
vector<int> v
;

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;

}

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: