Files
python/C++专题课程/背包九讲/TestProject/4_mixPackage.cpp

62 lines
1.6 KiB
C++
Raw Normal View History

2025-08-30 18:35:01 +08:00
#include <bits/stdc++.h>
using namespace std;
/**
* 4. 混合三种背包
1)问题:
有的物品只可以取一次(01背包),有的物品可以取无限次(完全背包),有的物品可以取的次数有一个上限(多重背包)。
应该怎么求解呢?
2)输入:
测试用例数
物品数 背包容量
第i种物品的ni(无限次的标为-1) ci wi
*/
#define maxV 1000
int f[maxV], v;
void ZeroOnePack(int ci, int wi) {
for (int j = v; j >= 0; j--)
if (j >= ci)
f[j] = max(f[j], f[j - ci] + wi);
}
void CompletePack(int ci, int wi) {
for (int j = 0; j <= v; j++)
if (j >= ci)
f[j] = max(f[j], f[j - ci] + wi);
}
void MultiplePack(int ni, int ci, int wi) {
if (ni * ci >= v) {
CompletePack(ci, wi);
return;
}
int k = 1, amount = ni;
while (k < ni) {
ZeroOnePack(ci * k, wi * k);
amount -= k;
k *= 2;
}
ZeroOnePack(ci * amount, wi * amount);
}
int main(void) {
int cases, n, ni, ci, wi;
freopen("../4.txt", "r", stdin);
cin >> cases;
while (cases--) {
memset(f, 0, sizeof(f));
cin >> n >> v;
for (int i = 0; i < n; i++) {
cin >> ni >> ci >> wi;
if (ni == 1) ZeroOnePack(ci, wi);
else if (ni == -1) CompletePack(ci, wi);
else MultiplePack(ni, ci, wi);
}
//for (int i = 0; i <= v; i++) cout << f[i] << " "; cout << endl;
cout << f[v] << endl;
}
return 0;
}