如何合并两个地图以保持更大的匹配键值？

Question

我在 Scala 中有两个可变映射。

val oldMap = Map(10 -> 100, 20 -> 200, 30 -> 300)
val newMap = Map(10 -> 101, 20 -> 200, 30 -> 299, 40 -> 400)

我想将newMap 合并到oldMap 以获得outputMap，如下所示，其中包含两个映射中的所有键，但值大于匹配键的值。

Map(20 -> 200, 40 -> 400, 10 -> 101, 30 -> 300)

我在下面尝试过，它可以工作，但我想知道 scala 的做法。

import scala.collection.mutable.Map

object Test extends App {

  val oldMap = Map(10 -> 100, 20 -> 200, 30 -> 300)
  val newMap = Map(10 -> 101, 20 -> 200, 30 -> 299, 40 -> 400)
  val outputMap = mergeMap(oldMap, newMap)

  println(outputMap)

  def mergeMap(map1: Map[Int, Int], map2: Map[Int, Int]): Map[Int, Int] = {
    val map1Keys = map2.keys
    val itr = map1Keys.iterator
    while (itr.hasNext)
    {
      val id = itr.next
      if (! map1.contains(id)){ // key not present in map1, INSERT
        map1(id) = map2(id)
      }
      else { // key present in map1, UPDATE
        if (map2(id) > map1(id)){
          map1(id) = map2(id)
        }
      }

    }
    map1
  }

  def commonMapKeys[A, B](a: Map[A, B], b: Map[A, B]): scala.collection.Set[A] = a.keySet.intersect(b.keySet)

}

Answer 1

您可以将地图变成元组的集合，合并它们，按键分组并选择组中的最大值：

val oldMap = Map(10 -> 100, 20 -> 200, 30 -> 300)
val newMap = Map(10 -> 101, 20 -> 200, 30 -> 299, 40 -> 400)
(oldMap.toSeq ++ newMap.toSeq)
  .groupBy(_._1)
  .mapValues(_.map(_._2).max)

或使用自 Scala 2.13 起可用的 groupMapReduce：

(oldMap.toSeq ++ newMap.toSeq)
    .groupMapReduce(_._1)(_._2)(math.max(_,_))

Answer 2

newMap.foldLeft(oldMap){ case (result, (k, newV)) =>
  val oldV = result.get(k)
  result + (k -> oldV.fold(newV)(newV max _))
}

您从oldMap 开始，并使用.foldLeft 遍历newMap。在每次迭代中，您从newMap 获取一个键和一个值，并尝试从结果映射（即开头的oldMap）中获取映射到相同键的值。您添加一个从 k 到新值或新旧值的最大值的新映射，以防旧映射也包含 k。

它与您拥有的几乎相同，只是以更实用的方式。