压缩稀疏序列

Question

我有两个对象序列，其中对象共享一个公共属性(让我们称之为id)。

case class ThingA(id: Int, someAttribute: String)
case class ThingB(id: Int, someOtherAttribute: Float)

我想加入“稀疏”列表，也就是说，每个ThingA.id都可能不匹配ThingB.id，反之亦然。in在每个列表中都是唯一的。

示例输入：

val thingAs = Seq(ThingA(0, "foo"), ThingA(1, "bar"))
val thingBs = Seq(ThingB(0, 1.0), ThingB(2, 0.3))

期望产出：

val zipped: Seq(Tuple[Option[ThingA], Option[ThingB]]) = Seq(
    (Some(ThingA(0, "foo")), Some(ThingB(0, 1.0))),   // Matching id = 0
    (Some(ThingA(1, "bar")), None),
    (None, Some(ThingB(2, 0.3))
)

我目前的尝试如下：

val zipped = (
    thingAs.map(a => (Some(a), thingBs.find(b => b.id == a.id))) ++
    thingBs.map(b => (thingAs.find(a => a.id == b.id), Some(b)))
).distinct

这很有效，但我希望有更好的方法。

Answer 1

一种方法是首先计算所有ids的联合，然后使用查找映射获取相应的ThingA或ThingB实例，就像@Dima建议的那样。

val lookupThingA = thingAs.map(x => x.id -> x).toMap
val lookupThingB = thingBs.map(x => x.id -> x).toMap

val zipped: Seq[(Option[ThingA], Option[ThingB])] =
    (lookupThingA.keySet | lookupThingB.keySet).map(i => (lookupThingA.get(i), lookupThingB.get(i))).toList

输出(请注意，您可以使用Set而不是Seq)：

(Some(ThingA(1,bar)),None)
(Some(ThingA(0,foo)),Some(ThingB(0,1.0)))
(None,Some(ThingB(2,0.3)))

(我让它作为一个社区维基，如果有人想要改善这一点)