Question

我正在研究《 Scala中的函数编程》一书，在数据结构这一章的最后，要求您根据filter来实现flatMap方法。以下是必要的功能和实现：

sealed trait List[+A]

case object Nil extends List[Nothing]

case class Cons[+A](head: A, tail: List[A]) extends List[A]

object List {
  def apply[A](as: A*): List[A] = {
    if (as.isEmpty) Nil
    else Cons(as.head, apply(as.tail: _*))
  }

  def append[A](l1: List[A], l2: List[A]): List[A] = {
    foldRight(l1, l2)((elem, acc) => Cons(elem, acc))
  }

  def concat[A](ls: List[List[A]]): List[A] = {
    foldLeft(ls, Nil: List[A])(append)
  }

  def map[A, B](l: List[A])(f: A => B): List[B] = {
    foldRight(l, Nil: List[B])((elem, acc) => Cons(f(elem), acc))
  }

  def filter[A](l: List[A])(f: A => Boolean): List[A] = {
    List.flatMap(l)(a => if (f(a)) List(a) else Nil)
  }

  def flatMap[A, B](l: List[A])(f: A => List[B]): List[B] = {
    concat(map(l)(f))
  }

  def foldRight[A, B](l: List[A], z: B)(f: (A, B) => B): B = {
    l match {
      case Nil => z
      case Cons(h, t) => f(h, foldRight(t, z)(f))
    }
  }

  def foldLeft[A, B](l: List[A], z: B)(f: (B, A) => B): B = {
    l match {
      case Nil => z
      case Cons(h, t) => foldLeft(t, f(z, h))(f)
    }
  }
}

实际的函数调用在这里：

val x = List(1, 2, 3, 4, 5)

List.filter(x)(_ < 3)

据我所知，在地图步骤之后，您将获得一个类似于以下内容的列表： Cons(Cons(1, Nil), Cons(2, Nil), Cons(Nil, Nil)...

我很难看到Nil的元素将从最终结果中滤除。

Answer 1

它们不是被“过滤掉”的。您将concat应用于列表列表后，它们便消失了，因为与空列表的连接无济于事。

麻烦flatMap如何消除Nil元素

1 个答案: