背包:01背包优化、完全背包、多重背包
01背包(ZeroOnePack): 有total件物品和一个容量为weight的背包。(每种物品均只有一件)第i件物品的费用是w[i],价值是v[i]。求解将哪些物品装入背包可使价值总和最大。
完全背包(CompletePack): 有total种物品和一个容量为weight的背包,每种物品都有无限件可用。第i种物品的费用是w[i],价值是v[i]。求解将哪些物品装入背包可使这些物品的费用总和不超过背包容量,且价值总和最大。
多重背包(MultiplePack): 有total种物品和一个容量为weight的背包。第i种物品最多有n[i]件可用,每件费用是w[i],价值是v[i]。求解将哪些物品装入背包可使这些物品的费用总和不超过背包容量,且价值总和最大。
比较三个题目,会发现不同点在于每种背包的数量,01背包是每种只有一件,完全背包是每种无限件,而多重背包是每种有限件。
01背包:
有N件物品和一个容量为V 的背包。放入第i件物品耗费的空间是Ci,得到 的价值是Wi。求解将哪些物品装入背包可使价值总和最大。
状态转移方程:dp[i][j]=max(dp[i-1][j],dp[i-1][j-w[i]]+v[i]);
#include <iostream>
#include <cstring>
#include <algorithm>
const int MAX = 1e3 + 5;
int w[MAX], v[MAX], dp[MAX];
using namespace std;
int main()
{
int total, weight;
cin >> total >> weight;
for (int i = 1; i <= total; i++){
cin >> w[i] >> v[i];
}
for (int i = 1; i <= total; i++){
for (int j = weight; j >= w[i]; j--){
dp[j] = max(dp[j], dp[j-w[i]] + v[i]);
}
}
cout << dp[weight] << endl;
return 0;
}
完全背包:
有N种物品和一个容量为V 的背包,每种物品都有无限件可用。放入第i种 物品的耗费的空间是Ci,得到的价值是Wi。求解:将哪些物品装入背包,可使 这些物品的耗费的空间总和不超过背包容量,且价值总和最大。
状态转移方程:dp[i][j]=max(dp[i-1][j],dp[i-1][j-k*w[i]]+k*v[i]);
与01背包的区别只有第二层for循环, 01背包从w[i]到weight,完全背包从weight到w[i]。因为k可以被优化掉。
#include <iostream>
#include <cstring>
#include <algorithm>
const int MAX=1e3+5;
int w[MAX],v[MAX],dp[MAX];
using namespace std;
int main()
{
int total,weight;
cin>>total>>weight;
for(int i=1;i<=total;i++)
{
cin>>w[i]>>v[i];
}
int k=1;
for(int i=1;i<=total;i++)
{
for(int j=w[i];j*k<=weight;j++)
{
dp[j]=max(dp[j],dp[j-k*w[i]]+k*v[i]);
}
k++;
}
cout<<dp[weight]<<endl;
return 0;
}
优化后:
#include <iostream>
#include <cstring>
#include <algorithm>
const int MAX = 1e3 + 5;
int w[MAX], v[MAX], dp[MAX];
using namespace std;
int main()
{
int total, weight;
cin >> total >> weight;
for (int i = 1; i <= total; i++){
cin >> w[i] >> v[i];
}
for (int i = 1; i <= total; i++){
for (int j = w[i]; j <= weight; j++){
dp[j] = max(dp[j], dp[j-w[i]] + v[i]);
}
}
cout << dp[weight] << endl;
return 0;
}
多重背包:
状态转移方程:f[i][j]=max(dp[i-1][j],dp[i-1][j-k*w[i]]+k*v[i](0<=k<=kmax));
有一种思路,将多件物品转换为多件数量为一的物品,例如 1 2 2变为 1 2 1 ,1 2 1
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAX=1e5+5;
int v[MAX],w[MAX],num[MAX],dp[MAX];
int main()
{
memset(dp,0,sizeof(dp));
int total,weight;
cin>>total>>weight;
for(int i=1;i<=total;i++)
{
cin>>w[i]>>v[i]>>num[i];
}
int k=total+1;
for(int i=1;i<=total;i++)
{
while(num[i]>1)
{
w[k]=w[i];
v[k]=v[i];
num[k]=1;
num[i]--;
k++;
}
}
for(int i=1;i<=k;i++)
{
cout<<w[i]<<" "<<v[i]<<" "<<num[i]<<" "<<endl;
}
for(int i=1;i<=k;i++)
{
for(int j=weight;j>=w[i];j--)
{
dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
}
}
cout<<dp[weight]<<endl;
return 0;
}