使用Scala中的Pattern Matching交换数组中的值

时间:2012-05-01 20:24:55

标签: arrays scala pattern-matching

我正在尝试从Scala for the impatient解决以下问题。问题如下:

  

使用模式匹配,编写一个函数交换,交换数组的前两个元素,前提是它的长度至少为两个。

我的解决方案是:

def swap(sArr:Array[Int]) = sArr.splitAt(2) match { 
                               case (Array(x,y),Array(z)) => Array(y,x,z)
                               case (Array(x,y),Array()) => Array(y,x)
                               case _ => sArr
                            }

我的问题在于第一个案例陈述。我认为它会匹配像(Array(1,2),Array(3))这样的东西,而我打算将它与模式匹配(Array(1,2),Array(3,4,5.....))

有人可以指出这是可能的。

谢谢

1 个答案:

答案 0 :(得分:9)

您的代码存在的问题是Array(z)表示“匹配单元素数组”。无论有多少元素,你想要的是z成为整个数组:

def swap(sArr: Array[Int]) = 
  sArr.splitAt(2) match { 
    case (Array(x, y), z) => Array(y, x) ++ z
    case _ => sArr
}

但是,我会用序列匹配语法_*编写它,这样你就不必手动拆分数组了:

def f(a: Array[Int]) = 
  a match {
    case Array(x, y, z @ _*) => Array(y, x) ++ z
    case _ => a
}