回溯法——01背包问题

问题 ：有n个重量为{w1,w2,wn}的物品，其价值是{v1,v2,vn}要求总重量为w，用0-1表示是否选择该物品，问如何选择价值最大

问题分析：这是一个求最优解问题 要求得一个序列x[n]数组 ,x[i]=0表示不取该物品，x[i]=1表示取。

深搜：

#include<iostream> 
#include<stdlib.h>
#include<algorithm>
using namespace std;

#define MAXN 100
int n = 4;//物品数量
int W = 6;//限制重量为6
int w[] = { 0,5,3,2,1 };
int v[] = { 0,4,4,3,1 };
int x[MAXN];//存放最终解
int maxv;//存放最优解的总价值

void Knapsack(int i, int tw, int tv, int op[]) {
	if (i > n) {//当前序号大于物品总量，说明找到一个叶子节点
		if (tw == W && tv > maxv) {
			maxv = tv;
			for (int i = 1; i < n; i++)
				x[i] = op[i];
		}
	}

	else {
		op[i] = 1;
		Knapsack(i + 1, tw + w[i], tv + v[i], op);
		op[i] = 0;
		Knapsack(i + 1, tw, tv, op);
	}
}

void display() {
	for (int i = 1; i <= n; i++) {
		cout << x[i];
	}
	cout << endl;
}

int main() {
	int op[MAXN];
	memset(op, 0, sizeof(op));
	Knapsack(1, 0, 0, op);
	display();
	return 0;
}

由最大价值可以剪枝 左孩子剪枝后的代码如下：

#include<iostream> 
#include<stdlib.h>
#include<algorithm>
using namespace std;

#define MAXN 100
int n = 4;//物品数量
int W = 6;//限制重量为6
int w[] = { 0,5,3,2,1 };
int v[] = { 0,4,4,3,1 };
int x[MAXN];//存放最终解
int maxv;//存放最优解的总价值

void Knapsack(int i, int tw, int tv, int op[]) {
	if (i > n) {//当前序号大于物品总量，说明找到一个叶子节点
		if (tw == W && tv > maxv) {
			maxv = tv;
			for (int i = 1; i < n; i++)
				x[i] = op[i];
		}
	}

	else {
		if (tw + w[i] <= W) {//左孩子节点剪枝
			op[i] = 1;
			Knapsack(i + 1, tw + w[i], tv + v[i], op);
		}
		op[i] = 0;
		Knapsack(i + 1, tw, tv, op);
	}
}

void display() {
	for (int i = 1; i <= n; i++) {
		cout << x[i];
	}
	cout << endl;
}

int main() {
	int op[MAXN];
	memset(op, 0, sizeof(op));
	Knapsack(1, 0, 0, op);
	display();
	return 0;
}

其实对右孩子也可以进行剪枝。我们知道当选择某件物品导致当前物品的总价值已经超过最大价值时就不用对此节点继续搜索了。但是当我们不选某件物品导致即使选取其后的所有物品都无法达到要求的重量，也可以进行剪枝，要进行右剪枝，需要在每一步中比较当前已选物品重量加上剩余物品重量之和与要求的重量大小。要做到这一点，需要在搜索函数中新添一个参数记录每一步的剩余重量。

#include<iostream> 
#include<stdlib.h>
#include<algorithm>
using namespace std;

#define MAXN 100
int n = 4;//物品数量
int W = 6;//限制重量为6
int w[] = { 0,5,3,2,1 };
int v[] = { 0,4,4,3,1 };
int x[MAXN];//存放最终解
int maxv;//存放最优解的总价值

void Knapsack(int i, int tw, int tv,int rw, int op[]) {//为了对有孩子进行剪枝，加入参数 剩余重量
	if (i > n) {//当前序号大于物品总量，说明找到一个叶子节点
		if (tw == W && tv > maxv) {
			maxv = tv;
			for (int i = 1; i < n; i++)
				x[i] = op[i];
		}
	}

	else {
		if (tw + w[i] <= W) {//左孩子节点剪枝
			op[i] = 1;
			Knapsack(i + 1, tw + w[i], tv + v[i],rw-w[i],op);
		}

		if (tw + rw > W) {
			op[i] = 0;
			Knapsack(i + 1, tw, tv, rw, op);
		}
		
	}
}

void display() {
	for (int i = 1; i <= n; i++) {
		cout << x[i];
	}
	cout << endl;
}

int main() {
	int op[MAXN];
	memset(op, 0, sizeof(op));
	Knapsack(1, 0, 0,W,op);
	display();
	return 0;
}