CDOJ 1136 邱老师玩游戏 树形01背包 (有依赖的背包问题)

点击打开链接

第一种思路：

建立森林，dp[i][k]表示森林中在第i颗树中选k个的最大价值 ， dp2[j]表示选j个的最大价值，对于每颗树，都更新一次。

转移方程：

dp[u][j] = max(dp[u][j],dp[u][j-k]+dp[v][k])；

dp2[j] = max(dp2[j],dp2[j-k]+dp[i][k])；

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;

const int maxn = 200+10;

vector<int> G[maxn];
int dp[maxn][maxn],dp2[maxn],a[maxn],va[maxn],n,m;

void dfs(int u){
dp[u][1] = va[u];
dp[u][0] = 0;
for(int i=0; i<G[u].size(); i++){
int v = G[u][i];
dfs(v);
for(int j=m; j>=2; j--)
for(int k=0; k<j; k++) // 已经选了u  所以不能再儿子里选全部j个 k<j
dp[u][j] = max(dp[u][j],dp[u][j-k]+dp[v][k]);
}
}

int main(){
while(scanf("%d%d",&n,&m)!=EOF){
if(n==0 && m==0) break;
memset(dp,0,sizeof(dp));
memset(dp2,0,sizeof(dp2));
for(int i=0;i<maxn;i++)
G[i].clear();
memset(a,0,sizeof(a));
for(int i=1; i<=n; i++){
cin >> a[i] >> va[i];
if(a[i]!=0)
G[a[i]].push_back(i); // 形成森林
}

for(int i=1; i<=n; i++){
if(a[i] == 0)
dfs(i); // 每一颗树都跑一遍01背包 dp[i][k]表示在第i颗树中选k个的最大价值
}

for(int i=1; i<=n; i++){
if(a[i] == 0){
for(int j=m; j>=0; j--)
for(int k=0; k<=j; k++)
dp2[j] = max(dp2[j],dp2[j-k]+dp[i][k]); // dp2[j]表示选j个的最大价值，对于每颗树，都更新一次
}
}

cout << dp2[m] << endl;
}
}

第二种思路：

先把依赖关系建成树，因为可能是森林，所以再建一个根节点把每个树的根节点连起来就可以。
然后树上跑dfs，在回溯的过程中跑01背包。
dp[u][i]表示在以u为根节点的子树上，选i个物品的最大价值。
然后转移方程就是dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k])，就是他可以从未更新的部分选j-k个，然后在已更新的那个儿子节点那里选k个，v是当前更新到的子节点，这样把k从1到j-1扫一遍，就把这个子节点v对父节点u做的贡献做出来了。
然后这里说一下为什么j要从m到2更新，因为我们要用到dp[u][j-k]，而这个代表的是未更新当前结点的情况，所以如果j从小更新的话，dp[u][j-k]就是已更新这个子节点的情况了，然后就错了，所以j要从大到小更新

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;

const int maxn = 200+10;

vector<int> G[maxn];
int dp[maxn][maxn],a[maxn],va[maxn],n,m;

void dfs(int u){
dp[u][1] = va[u];
dp[u][0] = 0;
for(int i=0; i<G[u].size(); i++){
int v = G[u][i];
dfs(v);
for(int j=m; j>=2; j--)
for(int k=0; k<j; k++)
dp[u][j] = max(dp[u][j],dp[u][j-k]+dp[v][k]);
// for(int k=1; k<=j; k++)
// 	dp[u][j] = max(dp[u][j],dp[u][k]+dp[v][j-k]); // u选k个 子节点中选j-k个
}
}

int main(){
while(scanf("%d%d",&n,&m),(n+m)){
memset(dp,0,sizeof(dp));
for(int i=0;i<maxn;i++)
G[i].clear();
memset(a,0,sizeof(a));
++m; // 虚拟根节点0， 一定选
for(int i=1; i<=n; i++){
cin >> a[i] >> va[i];
G[a[i]].push_back(i);
}

dfs(0);

cout << dp[0][m] << endl;
}
}