codevs 1155 金明的预算方案——来自深夜的挫败- -
- 没想到到了今天的最后时刻我还是没把这题做出来。最开始以为可能有3个附件,觉得自己枚举6种组合,代码太冗余了,就不想这么写。于是乎去写了一次错误的动态规划,原因是没有看清楚背包问题中,对于每个物品选或者不选的选择,它都会更次一次0-V的所有dp[i].
后来忍不住喵一下题解,看到了所谓的组合背包,我就傻傻地把每种组合当作一种新的物品用新的数组存起来,但是这样又犯了一个错误:得到的最优解可能包含同一主件的2种不同组合,不如主件<-附件1 和
主件<-附件2 , 这样的话主件就重复了,是不合法的。
去看了题解,才发现其实思路是那么的简单,代码是那么的简洁!只要在一次更新中选4种方案(包括都不选,就是5种)的1种就好了,平时模板的背包则是在1种(包含不选的话就是2种了)方案中选一种。也是自己第一次遇上组合背包。题解出处:http://codevs.cn/wiki/solution/?problem_id=1155
#include <iostream> #include <cstring> using namespace std; int f[40000],w[40000],v[40000],p[40000],s[40000][3]; int main() { int n,m,k=0; scanf("%d%d",&m,&n); for(int i=1;i<=n;i++) { scanf("%d%d%d",&w[i],&v[i],&p[i]); v[i]=v[i]*w[i];//直接将价值更改为重要度*重量 if(p[i]!=0) s[p[i]][++s[p[i]][0]]=i;//附件,指向主件 } for(int i=1;i<=n;i++) if(p[i]==0)//为主件再枚举 for(int j=m;j>=0;j--) { if(j-w[i]>=0) f[j]=max(f[j],f[j-w[i]]+v[i]); if(j-w[i]-w[s[i][1]]>=0&&s[i][1]!=0) f[j]=max(f[j],f[j-w[i]-w[s[i][1]]]+v[i]+v[s[i][1]]); if(j-w[i]-w[s[i][2]]>=0&&s[i][2]!=0) f[j]=max(f[j],f[j-w[i]-w[s[i][2]]]+v[i]+v[s[i][2]]); if((j-w[i]-w[s[i][1]]-w[s[i][2]])>=0) f[j]=max(f[j],f[j-w[i]-w[s[i][1]]-w[s[i][2]]]+v[i]+v[s[i][1]]+v[s[i][2]]);//上面枚举4种方案 } printf("%d",f[m]); return 0; }
oh~