动态规划解决背包问题

目录

一、需求

二、思路分析

三、代码实现

四、代码优化

上述代码的问题

path数组的遍历

优化后的代码

---

一、需求

        A:我现在有一张表，记录了商品名称，重量及价格;

        B:我还有一个给定容量的背包，这个背包用来存放商品;

        C:要求物品装入背包的价值最大，且重量不能超出;

        D:要求转入的物品不能重复(即01背包问题);

 二、思路分析

        A:定义两个一维数组来分别表示商品和背包容量;

        B:同样，需要一个二维数组v[i][j]来表示在背包容量为j时，前i个商品能装入背包的最大价值;

        C:根据A和B手动画出表格，然后根据表格分析规律;

 三、代码实现

package cn.itcast_01;

public class DynamicDemo {
	public static void main(String[] args) {
		// 物品的重量
		int[] w = { 1, 4, 3 };

		// 商品的价值
		int[] val = { 1500, 3000, 2000 };

		// 背包的容量
		int m = 4;

		// 商品的个数
		int n = val.length;

		// 创建二维数组
		// v[i][j]表示前i个商品中能够装入背包容量为j的最大价值
		int[][] v = new int[n + 1][m + 1];

		for (int i = 0; i < n + 1; i++) {
			v[i][0] = 0;
		}

		// 没有商品的情况
		for (int j = 0; j < m + 1; j++) {
			v[0][j] = 0;
		}

		// 根据规律写代码
		for (int i = 1; i < n + 1; i++) {
			for (int j = 1; j < m + 1; j++) {
				if (w[i - 1] > j) {
					v[i][j] = v[i - 1][j];
				} else {
					v[i][j] = Math.max(v[i - 1][j], val[i - 1] + v[i - 1][j - w[i - 1]]);
				}
			}
		}

		// 输出一下
		for (int i = 0; i < n + 1; i++) {
			for (int j = 0; j < m + 1; j++) {
				System.out.print(v[i][j] + " ");
			}
			System.out.println();
		}

	}
}

四、代码优化

• 上述代码的问题

        A:上述代码相当于把表格给打印出来了，但是并不知道是哪几个商品放入到了背包中使得价值最大;

        B:所以，需要定义一个二维数组path，来记录商品的存放情况，当商品加入背包并更新最大价值时，

           记录此时的path[i][j]=1；

• path数组的遍历

        A:若从0开始对path数组进行遍历，那么凡是path[i][j]=1的，都会输出，造成数据冗余;

        B:因此，尝试从后往前遍历，最大的价值肯定在最后;

• 优化后的代码

package cn.itcast_01;

public class DynamicDemo2 {
	public static void main(String[] args) {
		// 物品的重量
		int[] w = { 1, 4, 3 };

		// 商品的价值
		int[] val = { 1500, 3000, 2000 };

		// 背包的容量
		int m = 4;

		// 商品的个数
		int n = val.length;

		// 创建二维数组
		// v[i][j]表示前i个商品中能够装入背包容量为j的最大价值
		int[][] v = new int[n + 1][m + 1];

		// 定义二维数组记录商品的存放情况
		int[][] path = new int[n + 1][m + 1];

		for (int i = 0; i < n + 1; i++) {
			v[i][0] = 0;
		}

		// 没有商品的情况
		for (int j = 0; j < m + 1; j++) {
			v[0][j] = 0;
		}

		// 根据规律写代码
		for (int i = 1; i < n + 1; i++) {
			for (int j = 1; j < m + 1; j++) {
				if (w[i - 1] > j) {
					v[i][j] = v[i - 1][j];
				} else {
					// v[i][j] = Math.max(v[i - 1][j], val[i - 1] + v[i - 1][j -
					// w[i - 1]]);
					if (v[i - 1][j] < val[i - 1] + v[i - 1][j - w[i - 1]]) {
						v[i][j] = val[i - 1] + v[i - 1][j - w[i - 1]];
						// 把当前的情况记录下来
						path[i][j] = 1;
					} else {
						v[i][j] = v[i - 1][j];
					}
				}
			}
		}

		// 输出一下
		for (int i = 0; i < n + 1; i++) {
			for (int j = 0; j < m + 1; j++) {
				System.out.print(v[i][j] + " ");
			}
			System.out.println();
		}

		// 输出最后我们放入的是哪些商品
		int i = n;
		int j = m;
		while (i > 0 && j > 0) {
			if (path[i][j] == 1) {
				System.out.printf("第%d个商品放入到背包中\n", i);
				j -= w[i - 1];// 调整背包容量
			}
			i--;
		}
	}
}