如何计算SortedMap / TreeMap的运行总数

Question

所以我有下面的SortedMap

val mySortedMap: SortedMap[Double, Int] = SortedMap(1.1 -> 7, 2.4 -> 3, 6.5 -> 12)

现在我需要计算每个键的运行总数，所以输出应该如下所示：

val result: SortedMap[Double, Int] = SortedMap(1.1 -> 7, 2.4 -> 10, 6.5 -> 22)

我知道我可以用scanLeft做类似的事情

val result: Iterable[Int] = mySortedMap.scanLeft(0)((c, e) => c + e._2)

但是这会返回Iterable，而我需要保留上面指定的SortedMap。做这件事的最有效的方式是什么？

Answer 1

在累加器同时存在的情况下，可以使用foldLeft、一个新的SortedMap和正在运行的总计。

import scala.collection.immutable.SortedMap

def runningTotal(map: SortedMap[Double, Int]): SortedMap[Double, Int] = {
  val (transformed, _) = map.foldLeft((SortedMap.empty[Double, Int], 0)) {
    (acc, element) =>
      val (mapAcc, totalAcc) = acc
      val (key, value) = element
      val newTotal = totalAcc + value
      val newMap = mapAcc + (key -> newTotal)
      (newMap, newTotal)
  }
  transformed
}

更重要的是，这是您的解决方案，但是使用迭代器来代替，因此它会更高效一些。

def runningTotal(map: SortedMap[Double, Int]): SortedMap[Double, Int] = {
  val newValues = map.valuesIterator.scanLeft(0) {
    (acc, value) => acc + value
  }.drop(1)

  map.keysIterator.zip(newValues).to(SortedMap)
}

Answer 2

好吧，我想我可能找到了一个可能的解决方案：

val input: SortedMap[Double, Int] = SortedMap(1.1 -> 7, 2.4 -> 3, 6.5 -> 12)
val aggregates: Iterable[Int] = mySortedMap.scanLeft(0)((c, e) => c + e._2).tail
val sortedAggregates: SortedMap[Double, Int] = input.keySet.zip(aggregates).to(SortedMap)
println(sortedAggregates)

给TreeMap(1.1 -> 7, 2.4 -> 10, 6.5 -> 22)。

我想知道是否有更好的方法？