Scala模式匹配:如何匹配列表中的元素？

Question

是否可以使用Scala模式匹配重写以下代码？

val ls: List[String] = ??? // some list of strings

val res = if (ls.contains("foo")) FOO
     else if (ls.contains("bar")) BAR
     else SOMETHING_ELSE

Answer 1

您可以if为匹配添加条件,如下所示:

ls match {
  case x if x.contains("foo") => // FOO
  case x if x.contains("bar") => // BAR
  case _ => // ELSE
}

但是,这不是最好的方法,因为每个if检查都需要遍历列表,因此这不能很好地扩展.有各种不同的方法来处理这个问题,但我们需要更多地了解你的内涵,因为通常运行时语义会与你的代码不同(例如,你可以递归地遍历列表寻找"foo"或" bar",但这会假设你在列表中只有一个).

Answer 2

你可以使用像这样的函数来实现它

def onContains[T](xs: Seq[String], actionMappings: (String, T)*): Option[T] = {
  actionMappings collectFirst {
    case (str, v) if xs contains str => v
  }
}

并像这样使用它:

val x = onContains(items,
  "foo" -> FOO,
  "bar" -> BAR
)