背包问题总结

1.背包问题的定义

  背包问题是一种组合优化的NP完全问题。

  给定一个有限体积的背包一组物品,每个物品都有自己的重量价值,求背包可以装载物品的最大价值。

2.背包问题的分类

  背包问题根据每种物品可以装载的数量不同,可以分为以下几类:
01背包:每种物品只能装一次:
完全背包:每种物品可以装无限次
多重背包:每种物品可以装载的次数不同
混合背包:前面几种问题的组合
分组背包:给物品分组,每组物品只能选一个。

3.背包问题的求解

  背包问题一般使用动态规划求解,在使用动态规划时要明确4步:

  1. 确定dp数组的含义
  2. 确定递推公式
  3. 初始化dp数组
  4. 确定遍历的顺序

(1)01背包

  设背包能装载的最大重量是W,每件物品的重量是wight[i],价值是value[i],每个物品只有一个,问背包可以装载的物品的最大价值。
  首先分析问题,若用暴力解法,穷举所有情况,那么每件物品都有选或不选不选两种状态,计算下来就是 O ( n 3 ) O(n^3) O(n3)的时间复杂度,因为时间复杂度过高,所以一般用动态规划求解。

01背包的求解

第一步:确定dp数组

  一般使用二维数组dp[i][j]求解01背包问题,dp[i][j]表示在下标0-i的物品中选择,容量为j的背包可以装载的最大价值。

第二步:确定递推公式

  当选中当前物品时,因为放下i后当前容量不超过j,所以当前dp[i][j]的上一个状态容量为j - weight[i],这样在放下i后才能保证当前容量不超过j;当不选中当前物品时,上一个状态的容量最大可以还是j,因此递推公式为:
d p [ i ] [ j ] = m a x ( d p [ i − 1 ] [ j − w e i g h t [ i ] ] + v a l u e [ i ] , d p [ i − 1 ] [ j ] ) ( 1 ) dp[i][j] = max(dp[i - 1][j - weight[i]] + value[i], dp[i - 1][j]) (1) dp[i][j]=max(dp[i−1][j−weight[i]]+value[i],dp[i−1][j])(1)

第三步:初始化dp数组

  因为容量为0时可以装载的最大价值为0,所以dp[i]][0] = 0

第四步:确定遍历的顺序

  按公式(1),既可以现遍历i,也可以先遍历j,但是01背包中为了因为i只与i-1有关,可以用滚动数组优化,所以先遍历i。

初始Java代码

public static int test_01_bag(int volumn, int[] weight, int[] value) {
	int[][] dp = new int[weight.length][volumn.length + 1];
	for(int i = 1; i < weight.length; i++)
		for(int j = 1; j < = volumn; j++) {
			if(j < weight[i]) 
				dp[i][j] = dp[i - 1][j];
			else
				dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
		}
	return dp[weight.length - 1][volumn];
}
public class Main {
	public static void main(String[] args) {
		int[] weight = {1,2,4};
		int[] value = {30,35,40};
		int volumn = 6;
		int max_value = test_01_bag	(volumn, weight, value);
	}
}

优化Java代码

/*
因为i只和i-1有关,所以可以使用滚动数组优化
*/

public class Main {
	public static int test_01_bag_opt(int volumn, int[] weight, int[] value) {
	int[] dp = new int[volumn + 1];
	for(int i = 0; i < weight.length; i++)
		for(int j = volumn; j > = weight[i]; j--) {
			dp[j] = Math.max(dp[j], dp[j - weight[i]] + value[i]);
		}
	return dp[volumn];
}
	public static void main(String[] args) {
		int[] weight = {1,2,4};
		int[] value = {30,35,40};
		int volumn = 6;
		int max_value = test_01_bag_opt	(volumn, weight, value);
	}
}

组合问题

  组合问题和01背包有一定差别,组合问题是求有多少种装载方法,而不是求装载的最大价值。组合问题的递推公式一般为dp[i][j] = dp[i - 1][j] + dp[i - 1][j - weight[i]]。

经典例题

leetcode 416分割等和子集
leetcode 1049最后一块石头的重量2

(2)完全背包

  完全背包与01背包唯一不同的地方就是每种物品可以装载无限次。

完全背包的求解

  完全背包的求解和01背包相似,递推公式完全一样。不同的是遍历顺序
  在01背包中,如果使用二维dp,遍历顺序是任意的,如果使用一维dp,遍历顺序必须是先遍历物品,再遍历容量,且容量必须从大到小遍历。
  在完全背包中,如果使用一维dp,遍历顺序是随意的,但是对于容量必须从小到大遍历。

Java代码

public class Main {
	public static int test_complete_bag_opt(int volumn, int[] weight, int[] value) {
		int[] dp = new int[volumn + 1];
		for(int i = 0; i < weight.length; i++)
			for(int j = weight[i]; j <= volumn; j++) {
				dp[j] = Math.max(dp[j], dp[j - weight[i]] + value[i]);
			}
		return dp[volumn];
	}
	public static void main(String[] args) {
		int[] weight = {1,3,4};
		int[] value = {15,20,30};
		int volumn = 4;
		int max_value = test_complete_bag_opt(volumn, weight, value);
		System.out.println(max_value);
	}
}

完全背包的组合问题和排列问题

  如果是求组合问题,先遍历物品,再遍历容量;如果是求排列问题,先遍历容量,再遍历物品

经典例题

leetcode518:零钱兑换II
leetcode322:零钱兑换
leetcode377:组合总和IV
leetcode279:完全平方数

上一篇:使用python对bin文件进行操作


下一篇:字节面试算法题-0,1背包问题