动态规划之背包DP及其优化

avatar
作者
筋斗云
阅读量:0

(又是万恶的 DP ! ! ! )

OK啊,经过一个月的拖更,我又回来了!

请见这长~长~的目录:

跳过目录

目录

前言

01背包:

例题:洛谷P2871

状态设计:

转移方程:

代码:

优化:

优化代码:

完全背包:

例题:洛谷P1616​编辑

状态设计:

转移方程:

代码:

多重背包:

例题:洛谷P1776​编辑

状态设计:

优化:

二进制拆分:

代码:

单调队列:

代码:


跳过目录

前言

背包(Knapsack)问题是经典的动态规划问题,也很有实际价值。

而最经典的背包问题有三种:01背包,多重背包,完全背包。

01背包:

顾名思义,01背包,即每种物品只能取0件或1件。

例题:洛谷P2871

易证,考虑第i+1种物品时,可选择0件或1件。且答案只会利用前i种的利益最大值而与第i+2种选不选无关。(最优子结构+无后效性)

状态设计:

令f[i][j]为考虑前i个物品,容量为j的背包所能装的最大总价值。

转移方程:

取:f[i-1][j-w[i]]+v[i]

不取:f[i-1][j]

取max:f[i][j]=max(f[i-1][j],f[i-1][j-w[i]]+v[i])

代码:
#include <bits/stdc++.h> using namespace std; int f[3403][12881],w[114514],v[114514]; int main(){ 	int n, m; 	scanf("%d%d", &n, &m);     for(int i = 1; i <= n; i++){     	cin >> w[i] >> v[i]; 	}      for(int i = 1; i <= n; i++)     	for(int j = 0;j <= m; j++)     		f[i][j]=max(f[i-1][j],f[i-1][j-w[i]]+v[i]); 	printf("%d\n", f[n][m]); 	return 0; }
优化:

还能滚动数组优化空间为线性。

转移方程:f[i]=max(f[i],f[i-w]+v)

注意倒着更新即可。

优化代码:
#include <bits/stdc++.h> using namespace std; int f[114514]; int main(){ 	int n, m; 	scanf("%d%d", &n, &m); 	while(n--){ 		int w, v; 		scanf("%d%d", &w, &v); 		for(int i=m; i>=w; i--)f[i]=max(f[i], f[i - w] + v); 	} 	printf("%d\n", f[m]); 	return 0; }

完全背包:

例题:洛谷P1616
状态设计:

这种背包与01背包唯一的不同之处在于每个物品使用次数不限,所以参考01背包的dp状态:

 令f[i][j]为考虑前i个物品,容量为j的背包所能装的最大总价值。

转移方程:

一种十分显然的做法是每次输入时循环添加m/w[i]次。但时间复杂度明显超了。。。

完全背包与01背包唯一的不同之处在于每个物品使用次数不限。

想想是什么限制了01背包,使其每种物品只能取一个?​​​​​​​

真相只有一个!是它!

它保障了每次更新时如果当前位置取了物品,程序就不会再选取与当前一致的物品。

可以发现,实际上只需要将递推式改为f[i][j]=max(f[i-1][j],f[i][j-w[i]]+v[i])就能解决问题了。

而且类似的,它仍可以压缩空间为线性。

代码:

切记正序枚举!

​ #include <bits/stdc++.h> using namespace std; int f[114514]; int main(){ 	int n, m; 	scanf("%d%d", &n, &m); 	while(n--){ 		int w, v; 		scanf("%d%d", &w, &v); 		for(int i=w; i<=m; i++)f[i]=max(f[i], f[i - w] + v); 	} 	printf("%d\n", f[m]); 	return 0; }

多重背包:

例题:洛谷P1776
状态设计:

这种背包与01背包唯一的不同之处在于每个物品使用次数不只一次,所以参考01背包的dp状态:

 令f[i][j]为考虑前i个物品,容量为j的背包所能装的最大总价值。

一种十分显然的做法是每次输入时循环添加m[i]次。但时间复杂度明显超了。。。

(没错,就是前面复制的)

优化:
二进制拆分:

遇事不决就二分。

    ------XXX

显然log级别的复杂度能过。

考虑二进制拆分,即将原数拆分为若干个2^n和另一个数的和的形式。

例:

5 = ( 1 + 2 ) + 2
16 = ( 1 + 2 + 4 + 8 ) + 1
31 = ( 1 + 2 + 4 + 8 + 16 )

代码:
#include <bits/stdc++.h> using namespace std; int n, w, f[40005]; void add(int a, int b){ 	for(int i=w; i>=b; i--) 		f[i]=max(f[i], f[i - b] + a); } int main(){ 	scanf("%d%d", &n, &w); 	while(n--){ 		int a, b, c; 		scanf("%d%d%d", &a, &b, &c); 		for(int i=0; (1<<i)<=c; i++) 			add(a << i, b << i), c -= 1 << i; 		if(c) add(a * c, b * c); 	} 	printf("%d\n", f[w]); 	return 0; } 
单调队列:

上次关于它的一些基本知识我已经讲过了,不知道的看一看

优化:

可能有点抽象,多重背包选取物品实际上就是一个滑动窗口。

先看朴素做法的递推式:

f[i][j]=max{

f[i-1][j],

f[i-1][j-w[i]]+v[i],

f[i-1][j-2*w[i]]+2*v[i],

……

f[i-1][j-m[i]*w[i]]+m[i]*v[i]

}

可以发现当j1同余j2且其差不超过m[i]*w[i]时,f[i][j1]与f[i][j2]有部分重合。

容易想到滑动窗口。

对于每个i我们只要动态维护w[i]个长度为m[i]的单调队列即可。其内存储f[i-1][x+k*w[i]]-k*v[i]。

代码:
#include <bits/stdc++.h> using namespace std; int f[1005][20005], v[1005], w[1005], m[1005], q[20005]; int n, V; int main() {     cin >> n >> V;     for (int i = 1; i <= n; i++)         scanf("%d%d%d", v[i], w[i], m[i]);     for (int i = 1; i <= n; i++) {         for (int x = 0; x < v[i]; x++) {             int a = 0, b = -1;             for (int j = x; j <= V; j += v[i]) {//单调队列                 if (a <= b && q[a] < j - m[i] * v[i])a++;                 while (a <= b && f[i - 1][q[b]] - (q[b] - x) / v[i] * w[i] < f[i - 1][j] - (j - x) / v[i] * w[i])b--;                 q[++b] = j;                 f[i][j] = f[i - 1][q[a]] + ((j - q[a]) / v[i]) * w[i];             }         }     }     cout << f[n][V] << endl;     return 0; } 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!