回溯法解01背包问题

关于回溯法的概念,这篇文章讲的比较通俗易懂:https://blog.csdn.net/jarvischu/article/details/16067319

贴出阅读这篇文章后解决01背包问题的Java代码,注释中加入了自己的理解

package com.zxg.algorithm.backtrack;

/**
 * 回溯法解01背包问题
 * 背包问题的概念不再赘述。这里主要讲解下回溯法思路。
 * 将每一个物品分为装载和不装载两条路径,一个接一个的遍历,每遍历一个物品就会产生两条分支
 * 那么就会组成一棵树,深度遍历这棵树,找出最优解
 * 参考:https://blog.csdn.net/jarvischu/article/details/16067319
 */
public class PackageQuestion {

    public int[] weight;
    public int[] value;
    public int[] take;

    int curWeight = 0;
    int curValue = 0;

    int bestValue = 0;
    int[] bestChoice;
    int count;

    int maxWeight = 0;

    public void init(int[] weight, int[] value, int maxWeight) {
        if (weight == null || weight.length == 0
                || value == null || value.length == 0
                || weight.length != value.length || maxWeight <= 0) {
            System.out.println("args wrong!");
            return;
        }
        this.value = value;
        this.weight = weight;
        this.maxWeight = maxWeight;
        count = value.length;
        take = new int[count];
        bestChoice = new int[count];
    }

    public int[] maxValue(int x) {
        //走到了叶子节点
        if (x > count - 1) {
            //更新最优解
            if (curValue > bestValue) {
                bestValue = curValue;
                for (int i = 0; i < take.length; i++) {
                    bestChoice[i] = take[i];
                }
            }
        } else {
            //遍历当前节点(物品)的子节点:0 不放入背包 1:放入背包
            for (int i = 0; i < 2; i++) {
                take[x] = i;
                if (i == 0) {
                    //不放入背包,接着往下走
                    maxValue(x + 1);
                } else {
                    //约束条件,如果小于背包容量
                    if (curWeight + weight[x] <= maxWeight) {
                        //更新当前重量和价值
                        curWeight += weight[x];
                        curValue += value[x];
                        //继续向下深入
                        maxValue(x + 1);
                        //回溯法重要步骤,个人感觉也是精华所在。
                        // 当从上一行代码maxValue出来后,需要回溯容量和值
                        curWeight -= weight[x];
                        curValue -= value[x];
                    }
                }
            }
        }
        System.out.println(bestValue);
        return bestChoice;
    }

    public static void main(String[] args) {
        PackageQuestion question = new PackageQuestion();
        question.init(new int[]{7, 3, 4, 5},new int[]{42, 12, 40, 25},10);
        int[] result = question.maxValue(0);


    }
}

 

你可能感兴趣的:(数据结构与算法,回溯法,01背包问题)