回溯法解决0/1背包问题（c++）

0-1背包（回溯法）

题目内容

用回溯法解决0-1背包问题时，其解空间有长度为n的0-1向量组成，并可以组织成高度为n+1满二叉树。二叉树的结点按层次遍历的顺序从上到下、从左到右的顺序编号（从1开始）。

采用回溯法解决问题时，可以使用约束条件对左子树进行剪枝，使用限界条件对右子树进行剪枝。

在本题中，约束条件为tw+w[i]<=C，限界条件为tv+rv[i]>maxv

输入要求

有多组数据。

每组数据包含三行。

第一行包含两个整数C（1 <= C <= 10000）和n（1 <= n <= 20），分别表示背包的容量和物品的个数。

第二行包含n个整数，依次表示第i个物品的重量w（0 <= w[i] <= 100）。

第三行包含n个整数，依次表示第i个物品的价值v（0 <= v[i] <= 100）。

输出要求

对于每组输入数据

程序先依次输出扩展结点的编号、访问该结点时背包中物品的重量tw及背包中物品的价值tv，每个结点一行；

然后输出最优值maxv和最有解x（0/1向量）并换行。

解向量x的每个元素后一个空格，其他数据以空格分隔。

测试数据

输入
5 3
4 3 2
5 2 1
输出
1 0 0
2 4 5
5 4 5
11 4 5
5 1 0 0

参考答案

#include <iostream>
using namespace std;

void lp(int i, int tw, int tv, int op[],int n, int maxw, int& maxv, int weightArray[], int valueArray[], int maxArray[], int h)
{
	cout << h << " " << tw << " " << tv << "\n";
	if (i >= n)				
	{
		if ((tw <= maxw && tv > maxv) || tw < maxw && tv == maxv)		 
		{
			maxv = tv;
			for (int j = 0; j < n; j++)
			{
				maxArray[j] = op[j];
			}
		}
	}
	else				
	{
		if (tw + weightArray[i] <= maxw)
		{
			op[i] = 1;			//选取物品
			h = 2 * h;
			lp(i + 1, tw + weightArray[i], tv + valueArray[i], op, n, maxw, maxv, weightArray, valueArray, maxArray, h);
			h = h / 2;
		}
		int rv = 0;
		for (int d = i + 1; d < n; d++)
		{
			rv += valueArray[d];
		}
		if (tv + rv > maxv)
		{
			op[i] = 0;			//不选取物品 
			h = h * 2 + 1;
			lp(i + 1, tw, tv, op, n, maxw, maxv, weightArray, valueArray, maxArray, h);
			h = (h - 1) / 2;
		}

	}
}


int main()
{
	int maxw, n;
	while (cin >> maxw >> n)
	{
		int maxValue = 0;
		int* weightArray = new int[n];
		int* valueArray = new int[n];
		int* maxArray = new int[n];
		for (int i = 0; i < n; i++)
		{
			maxArray[i] = 0;
		}
		int* opArray = new int[n];
		for (int i = 0; i < n; i++)
		{
			cin >> weightArray[i];
		}
		for (int i = 0; i < n; i++)
		{
			cin >> valueArray[i];
		}
		int h = 1;
		lp(0, 0, 0, opArray, n, maxw, maxValue, weightArray, valueArray, maxArray, h);
		// 输出最优解
		cout << maxValue << " ";
		for (int i = 0; i < n; i++)
		{
			cout << maxArray[i] << " ";
		}
		cout << endl;

	}
	return 0;
}