Scala是否允许使用这种提取器?

时间:2011-07-15 20:02:03

标签: scala unapply

假设我有这个系列:

val a = Array(Array(1,2,3,4,5),Array(4,5),Array(5),Array(1,2,6,7,8))

有没有办法定义一个可以按以下方式工作的提取器:

a.foreach(e => {
   e match {
      case Array( ending with 5 ) => 
      case _ =>
   }
})

对于伪代码抱歉,但我不知道如何表达它。有没有办法匹配5个作为最后一个元素的东西?如果我想匹配第一个元素为1而最后一个元素为5的东西怎么办?这适用于各种长度的数组(请注意,我在示例中为我的数组专门选择了不同的长度。)

谢谢!

4 个答案:

答案 0 :(得分:11)

是的,你可以:

object EndsWith {
  def unapply[A]( xs: Array[A] ) = 
    if( xs.nonEmpty ) Some( xs.last ) else None
}

在你的例子上:

val a = Array(Array(1,2,3,4,5),Array(4,5),Array(5),Array(1,2,6,7,8))

a foreach { 
  case e @ EndsWith(5) => println( e.mkString("(",",",")" ) )
  case _ =>
}

按预期打印(1,2,3,4,5)(4,5)(5)

使用相同的方法,您可以编写一个提取器StartWith,然后添加一个方法,将它们组合在一个匹配两个条件的新提取器中。

答案 1 :(得分:9)

a.foreach(e => {
   e match {
      case a: Array[Int] if a.last == 5 => 
      case _ =>
   }
})

你可以做一些更好的事情来匹配第一个元素:

a.foreach(e => {
   e match {
      case Array(1, _*) => 
      case _ => 
   }
})

不幸的是@_*事件必须是数组参数列表中的最后一项。但是你可以在此之前进行匹配,因为它很复杂。

scala> val Array(1, x @_*) = Array(1,2,3,4,5)
x: Seq[Int] = Vector(2, 3, 4, 5)

scala> val Array(1, b, 3, x @_*) = Array(1,2,3,4,5)
b: Int = 2
x: Seq[Int] = Vector(4, 5)

答案 2 :(得分:4)

case语法支持if,因此可以使用:

a foreach {
  case a: Array[Int] if a.last == 5 =>
  case _ =>
}

答案 3 :(得分:1)

a.foreach (ar => ar.last match {                    
  case 5 => println ("-> 5] " + ar.mkString ("~"))
  case _ => println ("   ?] " + ar.mkString (":")) }) 

为什么不直接匹配最后一个元素?

-> 5] 1~2~3~4~5
-> 5] 4~5
-> 5] 5
   ?] 1:2:6:7:8