分组背包问题
题目描述
核心思路
最大价值应该是物品组i和背包容量j的函数。设f[i][j]
表示前i组物品,能放入容量为j的背包的最大价值
朴素算法应该是循环物品组,循环背包容量,对第 i i i组物品,容量为 j j j的背包,有 s + 1 s+1 s+1种选法,即 m a x ( f [ i − 1 ] [ j ] , f [ i − 1 ] [ j − v 1 ] + w 1 , f [ i − 1 ] [ j − v 2 ] + w 2 , ⋯ , f [ i − 1 ] [ j − v s ] + w s ) max(f[i-1][j],f[i-1][j-v_1]+w_1,f[i-1][j-v_2]+w_2,\cdots,f[i-1][j-v_s]+w_s) max(f[i−1][j],f[i−1][j−v1]+w1,f[i−1][j−v2]+w2,⋯,f[i−1][j−vs]+ws)
其中 f [ i − 1 ] [ j ] f[i-1][j] f[i−1][j]表示我不选择第 i i i组中的任何一个物品所得到的最大价值, f [ i − 1 ] [ j − v 1 ] + w [ i ] f[i-1][j-v_1]+w[i] f[i−1][j−v1]+w[i]表示我选择第 i i i组物品中的第一个物品所得到的最大价值,由于第 i i i组物品中的第一个物品用掉了 v 1 v_1 v1的体积,那么还剩下 j − v 1 j-v_1 j−v1的体积,因此我们从前 i − 1 i-1 i−1组物品中选择,那么最大价值就是 f [ i − 1 ] [ j − v 1 ] f[i-1][j-v_1] f[i−1][j−v1],然后加上选择第 i i i组中的第一个物品的价值 w [ i ] w[i] w[i]。因此总的最大价值就是 f [ i − 1 ] [ j − v 1 ] + w [ i ] f[i-1][j-v_1]+w[i] f[i−1][j−v1]+w[i]
可以对空间进行优化,将 f f f优化为一维
代码
没有优化空间, f f f是二维数组
/*
朴素算法:循环物品组,循环背包容量 对于第i组物品,容量为j的背包,有s+1种选法
max(f[i-1][j],f[i-1][j-V1]+W1,f[i-1][j-2V1]+W2,...,f[i-1][j-Vs]+Ws);
*/
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=110;
int v[N][N],w[N][N]; //体积 价值
//f[i][j]表示前i组物品,能放入容量为j的背包的最大价值
int f[N][N];
int s[N]; //s[i]=x表示第i组内有x件物品
int n,m; //物品 体积
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) //n组物品
{
scanf("%d",&s[i]); //第i组物品中有s[i]件物品
for(int j=1;j<=s[i];j++)
scanf("%d%d",&v[i][j],&w[i][j]);
}
for(int i=1;i<=n;i++) //n组物品
{
for(int j=1;j<=m;j++) //体积
{
for(int k=0;k<=s[i];k++) //决策 选择该组中的哪个物品
{
if(j>=v[i][k])
{
//max中的f[i][j]表示不选第i组中第k个物品,已获取的价值
//f[i-1][j-v[i][k]]+w[i][k]表示选入第i组物品中的第k个物品,能获取的价值
f[i][j]=max(f[i][j],f[i-1][j-v[i][k]]+w[i][k]);
}
}
}
}
printf("%d\n",f[n][m]);
return 0;
}
优化空间, f f f是一维数组
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=110;
int v[N],w[N]; //体积 价值
int f[N]; //将物品装入背包容量为m的最大价值
int n,m; //体积 价值
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) //物品
{
int s;
scanf("%d",&s);
for(int j=1;j<=s;j++)
scanf("%d%d",&v[j],&w[j]);
for(int j=m;j>=1;j--) //体积
{
for(int k=0;k<=s;k++) //决策 选择第i组中的第k件物品
{
if(j>=v[k])
f[j]=max(f[j],f[j-v[k]]+w[k]);
}
}
}
printf("%d\n",f[m]);
return 0;
}