分组背包问题,转化为0-1背包问题
购物单
http://www.nowcoder.com/questionTerminal/f9c6f980eeec43ef85be20755ddbeaf4
看了各个大佬的分析,终于写出了我自己的代码,其实知道了如何进行状态转移,就基本大差不差
#include <bits/stdc++.h>
using namespace std;
int main()
{
// 带条件的0-1背包问题
// 创建dp数组
int N(0), m(0);
cin >> N >> m;
// N是总预算,相当于背包问题的容量
int tv(0), tp(0), tq(0);
//vector<vector<int>> dp(m+1, vector<int>(N/10+1, 0));
vector<vector<int>> primary;
map<int, vector<vector<int>>> sub;
map<int, int> linenumtomainnum;
int i=0;
while (i<m) {
cin >> tv >> tp >> tq;
if (tq == 0)
{
primary.push_back(vector<int>{tv, tp, tq});
linenumtomainnum[i+1] = primary.size() - 1;
}
else {
sub[linenumtomainnum[tq]].push_back(vector<int>{tv, tp, tq});
}
i++;
}
// 有mainnum个主件
// 初始化
vector<vector<int>> w(primary.size(), vector<int>{});
vector<vector<int>> v(primary.size(), vector<int>{});
for (int i = 0; i<primary.size(); i++) {
int subsize = sub[i].size(); // 附件的数量
int wp = primary[i][0];
int vp = primary[i][0] * primary[i][1]; // 主件的w(价格)和v(价格乘重要程度)
// 加入只有主件的情况
w[i].push_back(wp);
v[i].push_back(vp);
if (subsize == 0) // 如果没有继续向下,说明存在至少一个附件
continue;
if (subsize == 1) // 只有一个附件
{
int ws1 = sub[i][0][0]; // 附件的价格
int vs1 = sub[i][0][0] * sub[i][0][1]; // 附件的价格乘重要程度
// 主件+附件的模式
w[i].push_back(ws1 + wp);
v[i].push_back(vs1 + vp);
}
if (subsize == 2) // 有两个附件,附件1+附件2
{
int ws1 = sub[i][0][0]; // 附件的价格
int vs1 = sub[i][0][0] * sub[i][0][1]; // 附件的价格乘重要程度
int ws2 = sub[i][1][0]; // 附件的价格
int vs2 = sub[i][1][0] * sub[i][1][1]; // 附件的价格乘重要程度
// 主件+附件1
w[i].push_back(wp + ws1);
v[i].push_back(vp + vs1);
// 主件+附件2
w[i].push_back(wp + ws2);
v[i].push_back(vp + vs2);
// 主件+附件1+附件2
w[i].push_back(wp + ws1 + ws2);
v[i].push_back(vp + vs1 + vs2);
}
}
// 动态规划开始
vector<vector<int>> dp(w.size() + 1, vector<int>(N / 10 + 2, 0));
for (int i = 1; i <= w.size(); i++) {
int ti = i - 1;
for (int j = 0; j <= N+10; j += 10) {
int tj = j / 10;
dp[i][tj] = dp[i - 1][tj];
for (int k = 0; k<w[ti].size(); k++) {
if(j>=w[ti][k])
dp[i][tj] = max(dp[i][tj], dp[i - 1][(j-w[ti][k])/10] + v[ti][k]);
}
}
}
cout << dp[w.size()][N/10+1] << endl;
return 0;
}不过我的代码没有经过优化,应该能用一维数组来做的,不过先不管了,菜鸡如我还是先学习背包问题的套路吧,优化后面再说


