【发布时间】:2017-05-30 11:08:34
【问题描述】:
您好,我正在尝试使用 Memoization 在 Scala 中实现背包解决方案。这是没有记忆的代码
// knapsack = maxWeight: Int, weights: List[Int], values: List[Int]
val knapsack: ((Int, List[Int], List[Int]) => Int) = Memo {
case (0, _, _) | (_, Nil, _) | (_, _, Nil) => 0
case (weight, headWt :: tailWts, _ :: tailVals) if headWt > weight =>
knapsack(weight, tailWts, tailVals)
case (weight, headWt :: tailWts, headVal :: tailVals) => math.max(
headVal+knapsack(weight-headWt, tailWts, tailVals),
knapsack(weight-headWt, tailWts, tailVals))
}
println(knapsack(50, weights.sortBy(-_), values.sortBy(-_)))
这是备忘录的定义:
case class Memo[A,B](f: A=>B) extends mutable.HashMap[A,B]{
override def apply(a: A):B = {
getOrElseUpdate(a, f(a))
}
}
但是,我得到一个编译时错误:
Type mismatch, expected: List[Int], actual: List[Any]
如果我删除备忘录,它会正常工作。
谢谢。
【问题讨论】: