Scala - 通过组合一个列表的内容和另一个列表的层次结构来创建一个新列表

Question

我正在尝试创建一个新列表，它将一个单级列表的内容与另一个嵌套列表的层次结构相结合，并通过两个列表中对象的 ID 将它们链接起来。注意：BChain 对象的 ID 存储在其中的 ItemObj 中。

我有三个对象：

ItemObj(id: String, stored: DateTime)
BChain(item: ItemObj, List[BChain])
AChain(aid: String, List[AChain])

我有两个列表：

val nestedList: List[AChain]
val singleLevelList: List[BChain]

我希望第二个列表具有第一个列表的层次结构，但仍包含它自己的元素。因此，BChain 在放入新需要的列表输出时，应该包括原始的ItemObj 和List[BChain] 属性（包括ItemObj 的所有原始属性数据-ID 和DateTime）。

所以代替（输入）：

 val nestedList = List(
 AChain("123", List(AChain("456", [])),
 AChain("789", [])
)

val singleLevelList = List(
 BChain(ItemObj("123", DateTime), []),
 BChain(ItemObj("456", DateTime), []),
 BChain(ItemObj("789", DateTime), []))
)

我想要以下输出：

val combinedLists = List(
     BChain(ItemObj("123", DateTime), List(BChain(ItemObj("456", DateTime), [])),
     BChain(ItemObj("789", DateTime), [])
)

作为最终名单。

注意：嵌套列表中的项目可能比单级列表中的项目多，如果是这种情况，则应忽略额外的项目。单级列表中的每一项都应对应于嵌套列表中的一项。

我怎样才能做到这一点？

任何帮助将不胜感激。

Answer 1

首先我必须模拟一个 DateTime 类型和值，以便编译以下内容。

case class ItemObj(id: String, stored: DateTime)
case class BChain(item: ItemObj, bcl :List[BChain])
case class AChain(aid: String, acl :List[AChain])

然后我将您的 singleLevelList 更改为真正的 Scala 代码并创建了一个 Map 以便快速查找。

val singleLevelList = List(
  BChain(ItemObj("123", DateTime), Nil),
  BChain(ItemObj("456", DateTime), Nil),
  BChain(ItemObj("789", DateTime), Nil)
)

val sLLMap = singleLevelList.groupBy(_.item.id)

接下来使用递归方法将所有AChains 更改为BChains。

def a2b(aLst :List[AChain]) :List[BChain] =
  aLst.map(a => BChain(sLLMap(a.aid).head.item, a2b(a.acl)))

现在来测试一下。

val nestedList = List(
  AChain("123", List(AChain("456", Nil))),
  AChain("789", Nil)
)

a2b(nestedList)  //appears to work

当然，如果nestedList 有一个 A 而singleLevelList 中没有对应的 B，这将抛出。

Answer 2

也许你想要这样的东西？

final case class ItemObj(id: String, stored: DateTime)
final case class BChain(item: ItemObj, list: List[BChain])
final case class AChain(aid: String, list: List[AChain])

def combine(as: List[AChain], bs: List[BChain]): List[BChain] = {
  val asMap = as.iterator.map(a => a.aid -> a.list).toMap

  def toBChain(a: AChain): BChain =
    BChain(
      item = ItemObj(id = a.aid, stored = ???),
      list = a.list.map(toBChain)
    )

  bs.map {
    case BChain(item, list) =>
      val newElements =
        asMap
          .getOrElse(key = item.id, default = Nil)
          .map(toBChain)

      BChain(
        item,
        list ::: newElements
      )
  }
}

Answer 3

即使嵌套列表的根元素在单级列表中不存在，此代码也将起作用

final case class ItemObj(id: String, stored: DateTime = null)
final case class AChain(aid: String, children: List[AChain] = List())
final case class BChain(bid: ItemObj, children: List[BChain] = List())

val nestedList = List(
  AChain("123", List(AChain("456"))),
  AChain("789")
)

def getCombineList(nestedList: List[AChain], singleList: List[BChain]): List[BChain] = {
  val singleListMap = singleList.groupBy(_.bid.id)

  def combine(items: List[AChain]): List[BChain] = {
    items flatMap {item => 
      val children = combine(item.children)
      val parent = singleListMap.get(item.aid).map(v =>
        List(v.head.copy(children = children))
      ).getOrElse(
        children
      )
      parent
    }
  }

  combine(nestedList)
}

val s = getCombineList(nestedList, List(BChain(ItemObj("456")), BChain(ItemObj("789"))))