HDU - 1561树形依赖背包

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;
}