1

我正在尝试在 scala 中创建一个按座位飞行的稀疏向量库，我遇到了 foldLeft 的问题，它似乎在长度为 1 的序列上创建或添加了一个额外的元素。

这是我的稀疏加法功能：

  def addTwoMaps(m1: Map[Int,Double], m2: Map[Int,Double]) =
  m1 ++ m2.map{ case (k,v) => k -> (v + m2.getOrElse(k, 0.)) }

这是我的“添加映射/稀疏向量序列并归一化”功能：

def addNMaps(ms : Map[Int, Double]*) = {
val denom = if (ms.length > 0) ms.length.toDouble else 1
ms.foldLeft(Map.empty[Int, Double])((a,b) => addTwoMaps(a,b)).mapValues(_ / denom)
}

（对于我的特殊情况，每个输入映射的值总和为 1，所以我所要做的就是除以参数序列的长度，以确保生成的映射总和为其值加一）

作为一个测试用例，如果我添加两个总和为 1 的地图，则效果很好：

scala> Common.addNMaps(Map(1->1), Map(1->1))
res34: scala.collection.immutable.Map[Int,Double] = Map(1 -> 1.0)

但如果我只有一个论点：

scala> Common.addNMaps(Map(1->1))
res33: scala.collection.immutable.Map[Int,Double] = Map(1 -> 2.0)

这些值突然加起来为两个！我的猜测是单曲Map(1->1)以某种方式被添加了两次，foldLeft但这只是一个猜测。

我究竟做错了什么？我怎样才能Common.addNMaps(Map(1->1))返回Map(1->1.0)？

4

1 回答 1

1

你的 有一个错字addTwoMaps，应该是：

def addTwoMaps(m1: Map[Int,Double], m2: Map[Int,Double]) =
  m1 ++ m2.map{ case (k,v) => k -> (v + m1.getOrElse(k, 0.)) }

你需要打电话getOrElse，m1而不是m2。

请注意，在这种情况下，您可以使用 an IntMap，它有一个方便的unionWith方法（可能是受到Haskell 的Data.Map.unionWith启发）：

import scala.collection.immutable.IntMap

def addNMaps(ms : IntMap[Double]*) = {
  val denom = if (ms.length > 0) ms.length else 1
   ms.foldLeft(IntMap.empty[Double]) {
     (a, b) => a.unionWith(b, (_, x, y) => x + y)
   }.mapValues(_ / denom)
 }

我不确定为什么unionWith它不是标准 Scala MapAPI 的一部分。

于 2012-05-17T13:55:22.880 回答