1. 二维费用背包
就和普通的01背包问题差不多;
#include<bits/stdc++.h>
using namespace std;
int f[105][105];
int main()
{
int n,v,m;cin>>n>>v>>m;
for(int i=0;i<n;i++){
int vv,mm,ww;cin>>vv>>mm>>ww;
for(int j=v;j>=vv;j--)
for(int k=m;k>=mm;k--)
f[j][k]=max(f[j][k],f[j-vv][k-mm]+ww);
}
cout<<f[v][m];
return 0;
}
2. 分组背包
OIWIKI定义:分组背包就是将物品分组,每组的物品相互冲突,最多只能选一个物品放进去。
因此分组背包的模板为
for(int i=0;i<n;i++)//n是组的编号的最大值
for(int j=m;j>;j--)//m是体积
for(int k=1;k<=b[i];k++)//b[i]是第i组的个数
f[j]=f[j-v[i][k]]+w[i][k];//状态转移方程
代码:【用普通的 二维数组+结构体 模拟两个vector好像更快一点】
#include<bits/stdc++.h>
using namespace std;
vector<int> v[1002],w[1002];int f[1004];
int main(){
cin.tie(0);ios::sync_with_stdio(0);
int n,m,x;cin>>m>>n;
for(int i=0;i<n;i++){
int vv,ww,kk;cin>>vv>>ww>>kk;v[kk].push_back(vv);w[kk].push_back(ww);
x=max(x,kk);
}
for(int i=1;i<=x;i++)
for(int j=m;j>=0;j--)
for(int k=0;k<v[i].size();k++){
if(j>=v[i][k])f[j]=max(f[j],f[j-v[i][k]]+w[i][k]);
}
cout<<f[m];
return 0;
}
例题:9. 分组背包问题
【十分潦草的代码】
#include<bits/stdc++.h>
using namespace std;
int f[104];int v[110][110];int w[110][110];int b[102];
int main(){
cin.tie(0);ios::sync_with_stdio(0);
int n,m;cin>>n>>m;
for(int i=1;i<=n;i++){
int x;cin>>b[i];
for(int j=1;j<=b[i];j++)cin>>v[i][j]>>w[i][j];
}
for(int i=1;i<=n;i++){
for(int j=m;j>=0;j--)
for(int k=1;k<=b[i];k++){
if(j>=v[i][k])f[j]=max(f[j],f[j-v[i][k]]+w[i][k]);
}
}
cout<<f[m];
return 0;
}