AcWing 10. 有依赖的背包问题 树形图的背包问题

AcWing 10. 有依赖的背包问题

有 N 个物品和一个容量是 V 的背包。

物品之间具有依赖关系，且依赖关系组成一棵树的形状。如果选择一个物品，则必须选择它的父节点。

如下图所示：
QQ图片20181018170337.png

如果选择物品5，则必须选择物品1和2。这是因为2是5的父节点，1是2的父节点。

每件物品的编号是 i，体积是 vi，价值是 wi，依赖的父节点编号是 pi。物品的下标范围是 1…N。

求解将哪些物品装入背包，可使物品总体积不超过背包容量，且总价值最大。

输出最大价值。

输入格式
第一行有两个整数 N，V，用空格隔开，分别表示物品个数和背包容量。

接下来有 N 行数据，每行数据表示一个物品。
第 i 行有三个整数 vi,wi,pi，用空格隔开，分别表示物品的体积、价值和依赖的物品编号。
如果 pi=−1，表示根节点。 数据保证所有物品构成一棵树。

输出格式
输出一个整数，表示最大价值。

数据范围
1≤N,V≤100
1≤vi,wi≤100
父节点编号范围：

内部结点：1≤pi≤N;
根节点 pi=−1;
输入样例
5 7
2 3 -1
2 2 1
3 5 1
4 7 2
3 6 2
输出样例：
11
这道题是树形问题的背包问题，这道题的难点在于怎么选取子节点，首先我们知道物品的价值是为wi[i]>0的，所以我们尽可能的去选这个物品。

我们先状态表达
$dp[i][j]$表示以i为根节点体积为j的最大价值。

第一步，我们选取这个物品，因为先选取这个物品才能选子节点的物品。然后开始分析，然后我们开始对从1-k体积开始放入。我们可以留给子节点选择的物品的大小为0-(m-vi[u])，然后我们开始对每个子节点进行一个放入操作。

最后代码如下：

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N=110;

int f[N][N];
//f[i][j]遍历到i节点物体个数为j。
//if (i不选) 则为0,选则分析子节点的最大值
int e[N],ne[N],h[N],idx;
int n,m;
int vi[N],wi[N];

void add(int a,int b)
{
    e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}

void dfs(int u)
{
    for(int i=h[u];~i;i=ne[i])
    {
        int son=e[i];
        dfs(son);
        for(int k=m-vi[u];k>=0;k--)
         for(int j=0;j<=k;j++)//为子背包留出j
          f[u][k]=max(f[u][k],f[u][k-j]+f[son][j]);
    }
    for(int i=m;i>=vi[u];i--) f[u][i]=f[u][i-vi[u]]+wi[u];
    for(int i=0;i<vi[u];i++) f[u][i]=0;
}
int main(void)
{
    memset(h,-1,sizeof h);
    cin>>n>>m;
    int root=-1;
    for(int i=1;i<=n;i++)
    {
        int father;
        cin>>vi[i]>>wi[i]>>father;
        if(father==-1)
        root=i;
        add(father,i);
    }
    dfs(root);
    cout<<f[root][m];
    
}