0-1背包问题回溯法

问题描述

给定n种物品和一背包。物品i的重量是Wi，其价值为Vi，背包的容量是C，问应如何选择装入背包中的物品，使得装入背包中物品的总价值最大。

分析

对于每一个物品有拿和不拿两种状态，所以对于第i+1个物品可以在第i个物品的后面分出两条路，一条拿，一条不拿，这样就构造出了一颗二叉树，我们只要搜索每一条路径找出最优解就行了，当背包能容纳当前物品时我们就取这个物品进入左子树，不能拿时就不进，右子树为不拿肯定能进，这样递归每一个物品就行了，直到递归到第n个结束，返回结果，在这2^n个结果中选取最优的保存就行

我们用bool类型数组cx记录当前物品的获取状态，用x记录最优解

代码

#include <iostream>
#include <algorithm>
using namespace std;
struct Node
{
    int id; // 序号
    int w;  // weight
    int v;  // value
    bool operator<(Node other)
    {
        return (other.v / other.w) < (v / w);
    }
};
int n;     // 物品数
int C;     // 背包容量
int bestP; // 当前最优价值
int cp;    // 当前价值
int cw;    // 当前重量
bool *x;   // 当前最优解
bool *cx;  // 当前解
Node *goods;
void backTrack(int i)
{
    if (i > n - 1)
    {
        if (bestP < cp)
        {
            for (int j = 0; j < n; j++)
            {
                x[j] = cx[j]; // 存储最优路径
            }
            bestP = cp;
        }
        return;
    }

    if (cw + goods[i].w <= C)
    {
        cw += goods[i].w;
        cp += goods[i].v;
        cx[goods[i].id] = true;     // 装入背包
        backTrack(i + 1); // 进入左子树
        cw -= goods[i].w;
        cp -= goods[i].v;
    }
    cx[goods[i].id] = false;    // 不装入背包
    backTrack(i + 1); // 回溯进入右子树
}
int main()
{
    cout << "物品种数n：";
    cin >> n;
    cout << "背包容量C：";
    cin >> C;
    x = new bool[n];
    cx = new bool[n];
    goods = new Node[n];
    for (int i = 0; i < n; i++)
    {
        cout << "第" << i + 1 << "件物品的重量和价值：";
        cin >> goods[i].w >> goods[i].v;
        x[i] = false;
        goods[i].id = i;
    }
    sort(goods, goods + n); //为了减少回溯的次数先排个序
    backTrack(0);
    cout << "装入背包的物品有：";
    for (int i = 0; i < n; i++)
    {
        if (x[i])
            cout << "第" << i + 1 << "件" << ' ';
    }
    cout << endl;
    cout <<"最大价值为：" <<bestP << endl;
    delete[] x;
    delete[] cx;
    delete[] goods;

    return 0;
}

输入

物品种数n：5
背包容量C：8
第1件物品的重量和价值：3 6
第2件物品的重量和价值：4 5
第3件物品的重量和价值：6 7
第4件物品的重量和价值：1 3
第5件物品的重量和价值：2 8

输出

装入背包的物品有：第1件 第4件 第5件
最大价值为：17