scala - 没有返回值的情况

Question

我有以下单元测试:

FlattenArray.flatten(
    List(0, 2, List(List(2, 3), 8, List(List(100)), null, List(List(null))), -2))
    should be(List(0, 2, 2, 3, 8, 100, -2))

我的实现如下:

object FlattenArray {
    def flatten(list: List[Any]): List[Any] = {
        list match {
            case Nil => Nil
            case (x: List[Any]) :: tail => flatten(x) ::: flatten(tail)
            case x :: tail => x :: flatten(tail)
        }
    }
}

测试如果失败，因为在 Nil 的情况下，我不应该向展平列表添加任何值:关于如何这样做有什么建议吗？

我可以从展平列表中过滤掉空值:这是正确的实现吗？

Answer 1

您可以为返回 flatten(tail) 的 null::tail 添加一个特例:

def flatten(list: List[Any]): List[Any] = {
  list match {
    case Nil => Nil
    case null :: tail => flatten(tail)
    case (x: List[Any]) :: tail => flatten(x) ::: flatten(tail)
    case x :: tail => x :: flatten(tail)
  }
}