地图合并功能的改进

Question

我正在编写一个函数来将两个 Map 合并在一起。这是我目前所拥有的：

def merge[K, V1, V2, V3](left: Map[K, V1], right: Map[K, V2])
    (fn: (Option[V1], Option[V2]) => V3): Map[K, V3] = {
  val r = (left.keySet ++ right.keySet) map {
    key =>
      (key -> fn(left.get(key), right.get(key)))
  }
  r.toMap
}

函数本身有效。你这样使用函数：

val m1 = Map(1 -> "one", 3 -> "three", 5 -> "five")
val m2 = Map(1 -> "I", 5 -> "V", 10 -> "X")
merge(m1, m2) { (_, _) } 
// returns: 
// Map(1 -> (Some(one),Some(I)), 
//     3 -> (Some(three),None), 
//     5 -> (Some(five),Some(V)), 
//     10 -> (None,Some(X)))

我有两个问题：

1. 我担心.get 和.toMap 调用的性能 计算复杂度。任何人都可以改进实施吗？
2. 我希望默认函数生成一对值 ({ (_, _) })。我无法完全掌握正确执行此操作的语法。

编辑： 虽然我最初说的是性能，但我指的是计算复杂度。 我的猜测是这个函数在 O(n•ln(n)) 时间内执行。 看起来我的函数大致在 O(n) 时间内执行。可以在 O(ln(n)) 内完成吗？

Answer 1

对于默认函数字面量使用：

(fn: (Option[V1], Option[V2]) => V3 = 
  (x: Option[V1], y: Option[V2]) => Tuple2(x,y))

你必须像这样使用合并：merge(m1,m2)()

我想说，在您对实际数据进行一些测量之前，不要担心性能。

编辑：关于性能，通过提供视图而不是构造地图，您可以快速“构造”而牺牲查找 - 假设我们正在处理不可变地图。因此，根据实际数据和用例，您可以获得更好的某些操作的性能，但它有一个权衡。

class MergedView[K, V1, V2, V3](
    left: Map[K, V1], right: Map[K, V2]
  )(fn: (Option[V1], Option[V2]) => V3 = (x: Option[V1], y: Option[V2]) => Tuple2(x,y)
  ) extends collection.DefaultMap[K, V3] {
  def get(key: K): Option[V3] = (left.get(key), right.get(key)) match {
    case (None, None) => None
    case t => Some(fn(t._1, t._2))
  }
  lazy val tuples = (left.keys ++ right.keys).map(key => key -> get(key).get)
  def iterator: Iterator[(K, V3)] = tuples.iterator
} 

val r1 = new MergedView(m1, m2)() // use parens here for second param list.

Answer 2

你不应该担心get——是的，它会创建一个包装器对象，但做其他任何事情都会很尴尬，除非分析器显示这是一个问题，否则你不应该尝试。

至于toMap，是的，这可能会让你慢下来。您可以尝试使用breakOut。

关于get和toMap的复杂度，对于不可变的HashMap，查找和添加是有效的常数时间，这是默认的Map。见Performance Characteristics of Scala Collections。