背包问题
0-1背包
dp问题思路
(一)状态表示(i,j)//i为物品个数,j为背包最大容量
1.集合
(1)所有选法
(2)条件
i.只从前i个物品里选
ii.选择物品总容量≤j
2.属性
Max,(Min,数量)
(二)状态计算
1. 集合划分f(i,j)
1.不含第i件物品(在1~i-1件物品里选)—f(i-1,j)
2.当j≥v[i]时 ,可能含第i件物品(在第1~i件物品里选,必须包含i)—f(i-1,j-v[i])+w[i]
f(i,j)=max( f(i-1,j) , f(i-1,j-v[i])+w[i]) )
2. 两个原则:
1.不重复
2.不漏选
每件物品最多装入一次
二维
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1010;
int v[N];
int w[N];
int dp[N][N];
int n,m;
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d",&v[i],&w[i]);
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(j>=v[i])dp[i][j]=max(dp[i-1][j],dp[i-1][j-v[i]]+w[i]);
else dp[i][j]=dp[i-1][j];
}
}
printf("%d",dp[n][m]);
return 0;
}
转化成一维
当计算f(i,j)是只用到了i-1且≤j,所以可以用滚动数组算,可将二维转化为一维:
删掉了第一维:在前i个物品中取。
f[j]表示:拿了总体积不超过j的物品,最大总价值。
为何能转化为一维?
二维时的更新方式:f[i][j]=max(f[i - 1][j] ,f[i - 1][j - v[i]] + w[i]);
1.对于每次循环的下一组i,只会用到i-1来更新当前值,不会用到i-2及之前值。于是可以在这次更新的时候,将原来的更新掉,反正以后也用不到。
所以对于i的更新,只需用一个数组,直接覆盖就行了。
2.对于每次j的更新,只需用到之前i-1时的j或者j-v[i],不会用到后面的值。
所以为了防止串着改,我们采取从后往前更新的方式,用原来i-1的数组来更新i。
(如果从前往后更新的话,前面的更新过之后,会接着更新后面的值,这样就不能保证是用原来i-1的数组来更新i的了)
如何转化为一维呢?
只用一个数组,每次都覆盖前面的数组。
1.如果当前位置的东西不拿的话,和前一位置的信息(原来i-1数组的这个位置上的值)是相同的,所以不用改变。
2.如果当前位置的东西拿了的话,需要和前一位置的信息(原来i-1数组的这个位置上值)取max。
所以,更新方式就为:
f[j]=max(f[j],f[j-v[i]]+w[i]);
整个更新方式就相当于:
每次i++,就从后往前覆盖一遍f数组,看每个位置上的值是否更新。
一维
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1010;
int v[N];
int w[N];
int dp[N];
int n,m;
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d",&v[i],&w[i]);
}
for(int i=1;i<=n;i++){
for(int j=m;j>=v[i];j--){
dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
}
}
printf("%d",dp[m]);
return 0;
}
完全背包问题
每件物品有无限个
dp问题思路
(一)状态表示(i,j)//i为物品个数,j为背包最大容量
1.集合
(1)所有选法
(2)条件
i.只从前i个物品里选
ii.选择物品总容量≤j
2.属性
Max
(二)状态计算
1. 集合划分f(i,j)
第i件物品可选择0~k个
(1)选择0个 f[i-1][j]
(2)选择k个
i.去掉k个物品i
ii.求Max f(i-1,j-kv[i])
iii.加上k个物品i f(i-1,j-kv[i])+k*w[i]
2. 两个原则:
1.不重复
2.不漏选
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1100;
int v[N],w[N];
int dp[N][N];
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d",&v[i],&w[i]);
}
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
for(int k=0;k*v[i]<=j;k++){
dp[i][j]=max(dp[i][j],dp[i-1][j-v[i]*k]+k*w[i]);
}
}
}
printf("%d",dp[n][m]);
return 0;
}
优化
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1100;
int v[N],w[N];
int dp[N][N];
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d",&v[i],&w[i]);
}
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
if(v[i]<=j)dp[i][j]=max(dp[i-1][j],dp[i][j-v[i]]+w[i]);
else dp[i][j]=dp[i-1][j];
}
}
printf("%d",dp[n][m]);
return 0;
}
发现与01背包很相似,可以利用滚动数组做进一步优化
与01背包不同的是dp[i][j]利用的是dp[i][j-v[i]]+w[i]进行更新,即在已更新的dp结果上进行更新,所以j由小到大
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1100;
int v[N],w[N];
int dp[N];
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d",&v[i],&w[i]);
}
for(int i=1;i<=n;i++){
for(int j=v[i];j<=m;j++){
dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
}
}
printf("%d",dp[m]);
return 0;
}
多重背包
每个物品最多有si个
问题思路
(一)状态表示(i,j)//i为物品个数,j为背包最大容量
1.集合
(1)所有选法
(2)条件
i.只从前i个物品里选
ii.选择物品总容量≤j
2.属性
Max
(二)状态计算
1. 集合划分f(i,j)
第i件物品可选择0~k个
(1)选择0个 f[i-1][j]
(2)选择k个
i.去掉k个物品i
ii.求Max f(i-1,j-kv[i])
iii.加上k个物品i f(i-1,j-kv[i])+k*w[i],k∈(0,si)
相当于滑动窗口最大值,必须用单调队列进行优化
朴素版
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1010;
int v[N],w[N],s[N];
int dp[N][N];
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d%d",&v[i],&w[i],&s[i]);
}
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
for(int k=0;k<=s[i]&&k*v[i]<=j;k++){
dp[i][j]=max(dp[i][j],dp[i-1][j-k*v[i]]+k*w[i]);
}
}
}
printf("%d",dp[n][m]);
return 0;
}
优化版
不能直接用完全背包的方式进行优化
已知前n-1个数最大值,求不出第n个数最大值
优化方法:二进制
0,1,2,4…2k 可以凑出0~2k+1-1的数
0,1,2,4…2k,c (c≤2k+1) 可以凑出0~s的数
用将原来第i个物品按1,2,4,…进行打包,用打包后物品表示新的物品
可以把O(n)一维线性问题优化为O(logn)
原问题时间复杂度变为O(NVlogS)
将多重背包问题转化为01背包问题
#include <iostream>
#include <algorithm>
using namespace std;
const int N=24000,M=2000;
int v[N],w[N];
int dp[N];
int main(){
int n,m;
int cnt=0;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
int a,b,s;
scanf("%d%d%d",&a,&b,&s);
int k=1;
while(k<=s){
cnt++;
v[cnt]=a*k;
w[cnt]=b*k;
s-=k;
k*=2;
}
if(s>0){
cnt++;
v[cnt]=a*s;
w[cnt]=b*s;
}
}
n=cnt;
for(int i=1;i<=n;i++){
for(int j=m;j>=v[i];j--){
dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
}
}
printf("%d",dp[m]);
return 0;
}
分组背包
物品分为n组,每组有若干种,枚举第i组物品最多选其中一个
第i组物品一个都不选dp[i-1,j]
第i组物品选第k个dp[i-1][j-v[i,k]]+w[i,k]
朴素版
#include <iostream>
#include <algorithm>
using namespace std;
const int N=110;
int v[N][N],w[N][N],s[N];
int dp[N][N];
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&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++){
for(int j=0;j<=m;j++){
for(int k=0;k<=s[i];k++){
if(j>=v[i][k])dp[i][j]=max(dp[i][j],dp[i-1][j-v[i][k]]+w[i][k]);
}
}
}
printf("%d",dp[n][m]);
return 0;
}
优化版
#include <iostream>
#include <algorithm>
using namespace std;
const int N=110;
int v[N][N],w[N][N],s[N];
int dp[N];
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&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++){
for(int j=m;j>=0;j--){
for(int k=0;k<=s[i];k++){
if(j>=v[i][k])dp[j]=max(dp[j],dp[j-v[i][k]]+w[i][k]);
}
}
}
printf("%d",dp[m]);
return 0;
}