【题目来源】AcWing 8. 二维费用的背包问题
【题意分析】
本题在前面背包问题的基础上,增加了一个维度——质量,背包拥有了容积、承重两个限制,物品也有了体积、质量两种属性。
【参考资料】第一讲 0/1背包问题
与0/1背包类似,加入一层循环限制重量即可:
#include<iostream>
using namespace std;
const int N1 = 1050;
const int N2 = 110;
struct node{
int v, m, w;
} q[N1];
int n, V, M;
int dp[N1][N2][N2];
//1000 * 100 * 100 = 107,1000ms能过
int main(){
cin >> n >> V >> M;
for(int i = 1; i <= n; i ++){
cin >> q[i].v >> q[i].m >> q[i].w;
for(int j = 0; j <= V; j ++){
for(int k = 0; k <= M; k ++){
dp[i][j][k] = dp[i - 1][j][k];
if(j >= q[i].v && k >= q[i].m){
dp[i][j][k]=max(dp[i][j][k],dp[i-1][j-q[i].v][k-q[i].m]+q[i].w);
}
}
}
}
cout << dp[n][V][M];
return 0;
}
同理,可以将该三维数组优化为二维数组:
for(int i = 1; i <= n; i ++){
for(int j = V; j >= q[i].v; j --){
for(int k = M; k >= q[i].m; k --){
dp[j][k] = max(dp[j][k], dp[j - q[i].v][k - q[i].m] + q[i].w);
}
}
}