0/1背包
背包问题是DP最经典的类型之一,而0/1背包是最经典最基础的背包问题。
背包体积为 V V V, n n n种物品,每种物品只有1个,第 i i i种物品对应体积为 c i c_i ci,价值为 w i w_i wi,怎样装填能使背包总价值最大?
由于每件物品只有选(0)与不选(1)两种情况,故称为0/1背包问题。
分析:闫氏DP分析法
- 状态表示
- 集合:定义数组 d p [ i ] [ j ] dp[i][j] dp[i][j],表示当前选取方案的价值。第 i i i行表示只考虑前 i i i个物品的放置情况, j j j表示当前选取体积不超过 j j j的方案集合。
- 属性: M a x Max Max
- 初始化:对于最值问题, d p [ i ] [ 0 ] = f [ 0 ] [ j ] = 0 dp[i][0]=f[0][j]=0 dp[i][0]=f[0][j]=0
- 状态计算:
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]:对于第
i
i
i种物品:
- 不可选第 i i i种物品: v < c [ i ] v<c[i] v<c[i],无法装入背包,背包剩余容积不变。集合状态仍为 [ 1 , i − 1 ] [1,i-1] [1,i−1],直接继承自第 i − 1 i-1 i−1种物品且背包容积仍为 j j j方案的价值。 d p [ i ] [ j ] = d p [ i − 1 ] [ j ] dp[i][j]=dp[i-1][j] dp[i][j]=dp[i−1][j]
- 可选第
i
i
i种物品:
- 不选第 i i i种物品:若选第 i i i种物品无法保证最优解,则不选,背包剩余容积不变。集合状态仍为 [ 1 , i − 1 ] [1,i-1] [1,i−1],直接继承自第 i − 1 i-1 i−1个物品且背包容积仍为 j j j方案的价值。 d p [ i ] [ j ] = d p [ i − 1 ] [ j ] dp[i][j]=dp[i-1][j] dp[i][j]=dp[i−1][j]
- 选第 i i i种物品:选第 i i i种物品可能导致产生最优解,则选。集合状态仍为 [ 1 , i − 1 ] [1,i-1] [1,i−1],因为0/1背包要求每种物品只能选一次,故继承自第 i − 1 i-1 i−1种物品且背包容积减少 c [ i ] c[i] c[i]方案的价值,并加 w [ i ] w[i] w[i]。 d p [ i ] [ j ] = d p [ i − 1 ] [ j − c [ i ] ] + w [ i ] dp[i][j]=dp[i-1][j-c[i]]+w[i] dp[i][j]=dp[i−1][j−c[i]]+w[i]
- 状态转移方程式: d p [ i ] [ j ] = m a x ( d p [ i − 1 ] [ j ] , d p [ i − 1 ] [ j − c [ i ] ] + w [ i ] ) dp[i][j]=max(dp[i-1][j],dp[i-1][j-c[i]]+w[i]) dp[i][j]=max(dp[i−1][j],dp[i−1][j−c[i]]+w[i])
遍历顺序:物品和背包谁先遍历都可以,根据状态转移方程式,dp数组的当前位置只与正上方和左上方有关,无论哪种遍历顺序,都可以确保在到达当前位置之前,正上方和左上方都有值。
初始化:
void init(){
for(int i=0;i<=n;i++) dp[i][0]=0;
for(int i=0;i<=v;i++) dp[0][j]=0;
}
void dp(){
for(int i=1;i<=n;i++)//遍历物品
for(int j=1;j<=v;j++)//遍历背包
if(c[i]<=j) dp[i][j]=max(dp[i-1][j],dp[i-1][j-c[i]]+w[i]);
else dp[i][j]=dp[i-1][j];
}
时间复杂度O( n v nv nv),空间复杂度O( n v nv nv)
DP表
滚动数组
DP问题的空间复杂度一般很高,可采用滚动数组方式对空间复杂度进行优化。
滚动数组原理是基于DP的无后效性,第 i i i行只与 i − 1 i-1 i−1行有关,至于 i − 1 i-1 i−1行之前的数据第 i i i行无需关注,因此在DP过程中实际上只有两行在进行工作,故可极大程度优化空间复杂度。
注意,滚动数组使中间信息丢失,若需要输出背包具体方案,则不能采用滚动数组。
交替滚动
思路:定义 d p [ 2 ] [ v ] dp[2][v] dp[2][v],当前工作指针 w o r k work work和上次工作指针 o l d old old,使用 d p [ w o r k ] [ v ] dp[work][v] dp[work][v]和 d p [ o l d ] [ v ] dp[old][v] dp[old][v]进行交替滚动,每次滚动后交换工作指针即可,思路简单
int dp[2][v];
void dp(){
int work=0,old=1;
for(int i=1;i<=n;i++){
swap(work,old);//交换工作指针而非交换数组元素
for(int j=1;j<=v;j++)
if(c[i]<=j) dp[work][j]=max(dp[old][j],dp[old][j-c[i]]+w[i]);
else dp[work][j]=dp[old][j];
}
}
自我滚动
思路:由状态转移方程式可知,当前元素只继承自上一行正上方( d p [ i − 1 ] [ j ] dp[i-1][j] dp[i−1][j])或上一行左上方( d p [ i − 1 ] [ j − 1 ] dp[i-1][j-1] dp[i−1][j−1]),因此逆序遍历背包容量进行更新,可将数组压至一维。
必须对背包进行逆序更新,这样是为了满足0/1背包每种物品只能选1个的性质,若顺序遍历则可能会对1种物品选多次,此时则为完全背包,且此错误必然会在选第1种物品时就发生。
自我滚动的0/1背包只可先遍历物品再遍历背包,不可颠倒(完全背包可颠倒)。
int dp[v];
void dp(){
for(int i=1;i<=n;i++){//顺序遍历物品
for(int j=v;j>=c[i];j--)//逆序遍历背包,装不下的不用管
dp[j]=max(dp[j],dp[j-c[i]]+w[i]);
}
}
输出具体方案
思路:定义标记数组,从 d p dp dp终点开始步步向上回溯,根据0/1背包状态转移方程式 p [ i ] [ j ] = m a x ( d p [ i − 1 ] [ j ] , d p [ i − 1 ] [ j − c [ i ] ] + w [ i ] ) p[i][j]=max(dp[i-1][j],dp[i-1][j-c[i]]+w[i]) p[i][j]=max(dp[i−1][j],dp[i−1][j−c[i]]+w[i])可知,判断 d p [ i ] [ j ] dp[i][j] dp[i][j]与 d p [ i − 1 ] [ j ] dp[i-1][j] dp[i−1][j]和 d p [ i − 1 ] [ j − c [ i ] ] + w [ i ] dp[i-1][j-c[i]]+w[i] dp[i−1][j−c[i]]+w[i]关系即可判断第 i i i个物品是否已装,最后输出标记数组。
注:求解具体方案仅适用于非滚动数组,因为滚动过程会将中间状态信息丢失。
extern int dp[MAX][MAX],i,j;//i,j:dp终点
bool f[MAX];
void print(){
for(;i>=1;i--){
if(j>=c[i]&&dp[i][j]==dp[i-1][j-c[i]+w[i]]){//说明第i个物品已选
f[i]=1;
j-=c[i];
}
}
for(int k=1;k<=n;k++) if(f[k]) cout<<k<<' ';
}