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