题目:
01背包问题
有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次
第 i 件物品的体积是 vi,价值是 wi
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大
输出最大价值
第一行两个整数,N,V,用空格隔开,分别表示物品数量和背包容积
接下来有 N 行,每行两个整数 vi,wi,用空格隔开,分别表示第 i 件物品的体积和价值
0 < N, V ≤ 1000
0 < vi, wi ≤ 1000
输入:
4 5
1 2
2 4
3 4
4 5
输出:
8
public class 动态规划_01背包 {
//v表示体积,w表示价值/权重,n和m分别表示物品数量和背包容积
public static int N = 10, n, m;
public static int[] v = new int[N], w = new int[N];
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
n = in.nextInt(); m = in.nextInt();
//先将数据读入到v和m中,切记我们是从1开始
for (int i = 1; i<=n; i++) {
v[i] = in.nextInt(); w[i] = in.nextInt();
}
base_01dp();
}
//基础版,二维数组表示状态
public static void base_01dp() {
int[][] f = new int[N][N];
//状态计算
for (int i = 1; i<=n; i++) { //枚举所有物品
for (int j = 0; j<=m; j++) { //枚举所有体积情况
//首先更新为不选当前i的情况,也就是f[i-1]层的状态
f[i][j] = f[i-1][j];
//然后考虑选择当前i的情况,只有当包含i的集合不空时候,再进行比较更新,因为如果为空,说明当前体积j装不下v[i]
if (j >= v[i]) f[i][j] = Math.max(f[i][j], f[i-1][j - v[i]] + w[i]);
}
}
System.out.println(f[n][m]);
}
//优化版,一维数组表示状态
public static void op_01dp() {
int[] f = new int[N];
//状态计算
for (int i = 1; i<=n; i++) {
//这里j的逆向处理是因为需要f[i-1]层的数据来计算f[i]层的状态
//而使用一维数组保存状态的话,从开头更新就会使得上一层的状态被提前更改,后续计算自然会出错
//为什么我们这里的终止条件是j>=v[i]? 而不是j>0? 因为我们只有当j装得下v[i]时才会进行更新
for (int j = m; j>=v[i]; j--) {
f[j] = Math.max(f[j], f[j-v[i]] + w[i]);
}
}
System.out.println(f[m]);
}
}
思路:
总体来说dp问题就是先尝试使用二维数组表示状态来进行解决,然后再通过分析题目来进行优化
基础版二维数组思路:
1.状态 f[i][j] 定义:前 i 个物品,背包容量为 j 下的最优解(最大价值)
2.当前背包容量不够(j < v[i]),没得选,因此前 i 个物品最优解即为前 i−1 个物品的最优解
3.当前背包容量够,可以选,因此需要决策选与不选第 i 个物品
优化版一维数组思路:
通过等价替换可以得到一维数组版本,也就是删除所有[i],而能够变为一维数组的前提是
1.每次计算 f[i] 这一层的状态时,最多只会用到 f[i-1] 这一层的状态,所以完全可以用一个滚动数组来代替二维数组
2.每次计算 f[i][j] 时,j 要么等于 j,要么是 j-v[i] 也就是小于 j,并没有超出当前 j 的范畴,所以可以用一维滚动数组来做
推荐视频讲解:
还在被背包问题所困扰吗?NOI金牌持有者y总亲情奉献,背包九讲你值得拥有:y总b站背包九讲专题
声明:
算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作学习记录和交流