编程之美中01背包问题的scala实现

问题描述：

有n个物体有重量和价值两个属性，一个能承重一定重量的背包。问怎么选择物体能实现背包里的价值最大化。比如假设有5个物体和一个背包。物体的重量分别是2、2、6、5、4，价值分别是6、3、5、4、6，背包承重为10。问怎么选择，能实现背包所背物体价值的最大化。

 

思路：

对于有T个物品，容量为V的情形，核心就是在容量为m时，将第n个物品放进来时的最大价值与不放进来时的最大价值相比较，取较大者。从第1个到第T个依次进行比较，从而得到不同容量下选取n个物品能够选取的最大价值二维数组opt，而opt[V][T-1]即为所求。具体实现如下：

class Package01 {

  def main(args: Array[String]) {
    val weights = Array(2,2,6,5,4)
    val values = Array(6,3,5,4,6)
    val length = weights.length
    val capacity = 10
    cal(weights,values,length,capacity)
  }

  def cal(weights: Array[Int], values: Array[Int], length: Int, capacity: Int): Unit ={
    assert(weights.length == values.length)
    assert(weights.length == length)
    assert(capacity>=0)

    //opt[i][j]  表示在容量为i时，在第0到j个物品中能够选取的最大价值
    val opt = Array.ofDim[Int](capacity+1,length)

    //将二维数组初始化
    (0 to length-1).map(
      x => {
        (0 to capacity).map(
          y => opt(y)(x) = 0
        )
      }
    )

    (0 to length-1).map(        //对每个物品做遍历
      j => {
        (0 to capacity).map(    //对每个容量做遍历
          i => {
            if(j>0){
              if(weights(j) <= i){
                //如果物品重量不高于包包容量，则比较取该物品与不取该物品的价值，取较大者
                opt(i)(j) = opt(i)(j-1)
                if(opt(i-weights(j))(j-1) + values(j) > opt(i)(j)){
                  opt(i)(j) = opt(i-weights(j))(j-1) + values(j)
                }
              }else{
                //如果物品重量高于包包容量，则不取该物品
                opt(i)(j) = opt(i)(j-1)
              }
            }else{
              //如果是第一个物品，则只需要将该物品重量与包包容量相比较即可
              if(i>=weights(j)) opt(i)(j) = values(j)
            }
          }
        )
      }
    )

    println(s"the max value is " + opt(capacity)(length-1))     //打印最大价值
    println(opt.map(_.mkString(",")).mkString("\n","\n","\n"))  //打印opt二维数组
  }

}

经过测试，结果打印如下：
the max value is 15

0,0,0,0,0
0,0,0,0,0
6,6,6,6,6
6,6,6,6,6
6,9,9,9,9
6,9,9,9,9
6,9,9,9,12
6,9,9,10,12
6,9,11,11,15
6,9,11,13,15
6,9,14,14,15