如何合并两个元组列表？

Question

我在Scala中有两个列表，如何合并它们以便将元组组合在一起？

是否有现成的Scala列表API可以执行此操作或需要我自己执行此操作？

输入：

 List((a,4), (b,1), (c,1), (d,1))
 List((a,1), (b,1), (c,1))

预期产量：

List((a,5),(b,2),(c,2),(d,1))

Answer 1

您可以尝试以下一行：

scala> ( l1 ++ l2 ).groupBy( _._1 ).map( kv => (kv._1, kv._2.map( _._2).sum ) ).toList
res6: List[(Symbol, Int)] = List(('a,5), ('c,2), ('b,2), ('d,1))

其中l1和l2是要合并的元组列表。

现在，细分：

• (l1 ++ l2)你只是连接两个列表
• .groupBy( _._1)您按照第一个元素对所有元组进行分组。 您将收到一个Map，第一个元素作为键，以及以此元素作为值开头的元组列表。
• .map( kv => (kv._1, kv._2.map( _._2).sum ) )你制作一个新的地图，使用相似的键，但这些值是所有第二个元素的总和。
• .toList将结果转换回列表。

或者，您可以使用模式匹配来访问元组元素。

( l1 ++ l2 ).groupBy( _._1 ).map{
  case (key,tuples) => (key, tuples.map( _._2).sum ) 
}.toList

Answer 2

或者，您也可以使用mapValues来缩短代码。

正如您可能猜到的， mapValues允许您仅重新映射groupBy创建的Map中每个（键，值）对的值。

在这种情况下，传递给mapValues的函数将每个（Char，Int）元组mapValues为Int，然后将得到的Int列表相加。

(l1 ::: l2).groupBy(_._1).mapValues(_.map(_._2).sum).toList

如果输出列表的顺序需要遵循您的示例，则只需添加依赖于Ordering [（Char，Int）]隐式实例的sorted 。

(l1 ::: l2).groupBy(_._1).mapValues(_.map(_._2).sum).toList.sorted

Answer 3

如果你可以假设List[(A,B)]都是按照Ordering[A] ，你可以这样写：

def mergeLists[A,B](one:List[(A,B)], two:List[(A,B)])(op:(B,B)=>B)(implicit ord:Ordering[A]): List[(A,B)] = (one,two) match {
    case (xs, Nil) => xs
    case (Nil, ys) => ys
    case((a,b)::xs,(aa,bb)::ys) =>
      if (a == aa) (a, op(b,bb)) :: mergeLists(xs,ys)(op)(ord)
      else if (ord.lt(a,aa)) (a, b) :: mergeLists(xs, (aa,bb)::ys)(op)(ord)
      else (aa, bb) :: mergeLists((a,b) :: xs, ys)(op)(ord)
}

不幸的是，这不是尾递归。