在 scala.immutable.Map 中更新多个值的惯用方法

Question

我需要更新（检索和增加）绑定到映射中两个键的两个不同值。这两个键有时会重合。 我现在有以下代码：

// val map: Map[Int, Int]
// val key1, key2: Int
if (key1 == key2) {
  tailRecFunction(someArg, map 
    + Tuple2(key1, 2 + map.getOrElse(key1, 0)))
} else {
  tailRecFunction(someArg, map 
    + Tuple2(key1, 1 + map.getOrElse(key1, 0))
    + Tuple2(key2, 1 + map.getOrElse(key2, 0)))
}

如你所见，如果你在key1 == key2 时使用else 块，那么key1 == key2 的值将错误地增加1 而不是2 --- 第二个元组错误地更新原始值，而不是第一个元组应用的值。
有没有更简洁的写法？

Answer 1

首先，您可以简化映射以在不存在键时返回0：

val map0 = Map.empty[Int, Int] withDefaultValue 0

那么您可以放心地调用map(key) 而不是map.getOrElse(key, 0)。

其次，您可以使用箭头关联语法来创建Tuple2 实例。 IE。 key -> value 而不是 Tuple2(key, value)。

第三，我根本不会介绍if ... then ... else。只需连续更新每个键的映射：

def addKey(map: Map[Int, Int], key: Int) = map + (key -> (map(key) + 1))

val map1 = addKey(map0, key1)
val map2 = addKey(map1, key2)
tailRecFunction(someArg, map2)

Answer 2

另一个SO Answer 建议使用 Map (ScalaZ) 的 Monoid 特性：

tailRecFunction(someArg, map |+| Map(key1 -> 1) |+| Map(key2 -> 1))

|+| 运算符将对属于同一键的值求和。