HDU - 1561树形依赖背包
题目: 传送门
思路见代码注释.
代码:
/*
对于每个节点u,要么有唯一的父亲fa,要么没有父亲,自形一颗树
所以按照题目给的要求该图是一个森林.
我们考虑将森林中的每一颗树的根连接到超级根root.那么形成的图是一个以超级根
为根的树,题目也转化为从该树上取m+1个节点,使得价值最大,取u节点必须先取父亲节点.
这就是树形依赖背包的问题了
*/
#include<bits/stdc++.h>
#define mset(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int inf=0x3f3f3f3f;
const int N=1e4+10;
const int mod=1e9+7;
int w[305];
vector<int> g[305];
int dp[305][305];
int n,m;
void dfs(int u)
{
for(int v:g[u])
{
dfs(v);
/*因为要求该儿子中要么取,要么不取, 所以我们必须先枚举背包容量从大到小,其次枚举儿子的容量大小*/
for(int s=m;s>=1;--s)//枚举背包容量从大到小
{
for(int a=1;a<=s;++a){//其次枚举儿子容量
dp[u][s]=max(dp[u][s],dp[u][s-a]+dp[v][a]);
}
}
}
dp[u][0]=0;
for(int i=m;i>=1;--i)
dp[u][i]=dp[u][i-1]+w[u];
}
int solve()
{
dfs(0);
return dp[0][m];
}
int main()
{
ios::sync_with_stdio(false);cin.tie(0);
while(cin>>n>>m,n)
{
for(int i=0;i<=n;++i) g[i].clear();
for(int i=1;i<=n;++i){
int fa,ww;
cin>>fa>>ww;
w[i]=ww;
g[fa].push_back(i);
}
++m;//0这个节点额外的且是必须取的,所以m要+1
mset(dp,0);
cout<<solve()<<endl;
}
return 0;
}