（Scala）在输出中保留列表的结构

Question

我需要一点帮助。

我一直在研究 Scala 的功能方面。几乎所有的工作都在列表中，在大多数情况下，我可以解决问题，但我遇到了一个小死胡同。在某些问题中，我无法保留列表的原始形式（结构）。在输出中，所有嵌套列表都被展平。连接 ::: 使列表变平（即原始列表的一个元素）并附加 :: 给我一个编译错误，因为它需要通用 T 类型（不是列表）。

一个非常简单的例子，我想删除列表中与输入匹配的第一个元素：

def removeFirst[T](obj: T, list: List[T]): List[T] = {
  if (list isEmpty) Nil
  else{
  val fin: List[T] = list.head match {
    case headAsList: List[T] => if (containsWithNestedLists(obj, headAsList))
      removeFirst(obj, headAsList) ::: list.tail
    else headAsList ::: removeFirst(obj, list.tail)
    case _ => if (list.head == obj) list.tail
    else if (list.tail == List()) List(list.head)
    else list.head :: removeFirst(obj, list.tail)
  }
  fin
  }
}

对于一级深度列表，它可以正常工作，但是 出来的输出 removeFirst(1,List(List(1,2,3),1,2,3,4,5,6)) 是 List(2, 3, 1, 2, 3, 4, 5, 6)，理想情况下我想要 List(List(2,3),1, 2,3,4,5,6)).

或者更具体的输入removeFirst(1,List(List(2,3,List()),List(1,2,3),1,2,3,4,5,6,List(2,3,List()))) 应该有输出=List(List(2,3,List()),List(2,3),1,2,3,4,5,6,List(1,2,3,List()))

我还发现删除通用 T 并在其位置使用 Any 可以解决问题，但我也知道 Any 是一个很大的 no-no 和临时解决一个永久性问题，因为它在其他功能中没有帮助。

据我所知，我还没有在互联网上看到有用的解决方案，所以我不得不问。我错过了什么，需要调试还是有其他功能可以帮助我？我最接近答案的方法是以某种方式使用 append ::，但我可能错了。

Answer 1

您的示例看起来好像您想从列表中删除某个元素（如果存在）。 对于平面列表，您可以更简单地执行此操作：

def removeFirst[T](obj: T, list: List[T]) = list match {
  case `obj` :: rest => rest
  case _ => list
}

这将执行以下操作：

> removeFirst(1, List(1, List(1,2,3)))
res57: List[Any] = List(List(1, 2, 3))
> removeFirst(1, List(2, List(1,2,3)))
res58: List[Any] = List(2, List(1, 2, 3))
> removeFirst(List(2,3), List(List(2,3), List(1,2,3)))
res59: List[List[Int]] = List(List(1, 2, 3))

但是，您似乎想对任意嵌套的列表执行此操作。这不是直接可能的，因为 Scala 无法表达它的确切类型。您需要的类型类似于

type NestedList[T] = List[T union NestedList[T]]

Scala 没有联合类型，你不能以这种方式进行递归定义，所以你也不能这样做：

type NestedList[T] = List[Either[T, NestedList[T]]] 

但是，如果您使用类而不是类型，则可以这样做：

case class NestedList[T](value: List[Either[T, NestedList[T]]])

现在你可以这样写你的算法了：

def removeFirst[T](obj: T, list: NestedList[T]): NestedList[T] = {
  val rest = list.value match {
    case Left(`obj`) :: tail => tail
    case __ => list.value
  }
  NestedList(rest.map {
    case Right(r) => Right(removeFirst(obj, r))
    case Left(r) => Left(r)
  })
}

你可以这样做：

> removeFirst(1, NestedList(List(Left(1), Left(2),  Right(NestedList(List(Left(1),Left(3)))))))
res71: NestedList[Int] = NestedList(List(Left(2), Right(NestedList(List(Left(3))))))

构建和分解这些结构当然有点麻烦。所以也许使用密封的抽象类和两个案例类而不是使用 Either 来构建一个合适的树类会更好。