我正在尝试从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.....))
有人可以指出这是可能的。
谢谢
答案 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
}