回溯法解决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;
}
  • 1
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值