AcWing上的一道混合背包裸题用到了多重背包的二进制拆分为01背包,以及STL的vector
链接:https://www.acwing.com/problem/content/7/
- 混合背包问题
有 N 种物品和一个容量是 V 的背包。
物品一共有三类:
第一类物品只能用1次(01背包);
第二类物品可以用无限次(完全背包);
第三类物品最多只能用 si 次(多重背包);每种体积是 vi,价值是wi。
求解将哪些物品装入背包,可使物品体积总和不超过背包容量,且价值总和最大。
输出最大价值。
输入格式
第一行两个整数,N,V,用空格隔开,分别表示物品种数和背包容积。接下来有 N 行,每行三个整数 vi,wi,si,用空格隔开,分别表示第 i 种物品的体积、价值和数量。si=−1 表示第 种物品只能用1次;si=0表示第 i 种物品可以用无限次;si>0表示第 i 种物品可以使用 si 次;
输出格式
输出一个整数,表示最大价值。
数据范围
0<N,V≤1000
0<N,V≤1000
0<vi,wi≤1000
−1≤si≤1000
输入样例
4 5
1 2 -1
2 4 1
3 4 0
4 5 2
输出样例:
8
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int N = 1010;
int n, V, v, w, s;
int dp[N];
struct Thing{//定义结构体
int w, v, kind;
};
vector<Thing>things;
int main(){
cin >> n >> V;
for (int i = 0; i < n; i++){
cin >> w >> v >> s;//w体积,v价值,s类型
if (s < 0) things.push_back({ w, v, -1 });//01背包
else if (s == 0) things.push_back({ w, v, 0 });//完全背包
else{//将多重背包二进制拆分转化为01背包
for (int k = 1; k <= s; k *= 2){
s -= k;
things.push_back({ w*k, v*k, -1 });
}
if (s > 0) things.push_back({ w*s, v*s, -1 });
}
}
for (auto thing : things){//相当于auto it=......
if (thing.kind < 0){//01
for (int j = V; j >= thing.w; j--)
dp[j] = max(dp[j], dp[j - thing.w] + thing.v);
}
else{//完全
for (int j = thing.w; j <= V; j++)
dp[j] = max(dp[j], dp[j - thing.w] + thing.v);
}
}
cout << dp[V] << endl;
return 0;
}