二维费用背包&&分组背包

1. 二维费用背包

就和普通的01背包问题差不多；

例题： 8. 二维费用的背包问题 - AcWing题库

#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];//状态转移方程
            

例题：P1757 通天之分组背包 

代码：【用普通的 二维数组+结构体 模拟两个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;
}