Question

我想以功能样式重构这段 Scala 代码：

var k = -1
for (i <- 0 until array.length)
  if ((i < array.length - 1) && array(i) < array(i + 1))
    k = i

Scala 中的数组有indexWhere，可用于val index = array.indexWhere(c => c == 'a')之类的内容。我正在寻找类似的东西，它会考虑数组的两个连续元素。

Answer 1

当你需要查看集合中的相邻元素时，通常的功能方法是用它的尾部“压缩”集合。请考虑以下简化示例：

scala> val xs = List(5, 4, 2, 3, 1)
xs: List[Int] = List(5, 4, 2, 3, 1)

scala> val tail = xs.tail
tail: List[Int] = List(4, 2, 3, 1)

scala> xs.zip(tail)
res0: List[(Int, Int)] = List((5,4), (4,2), (2,3), (3,1)

现在我们可以使用indexWhere：

scala> res0.indexWhere { case (x, y) => x < y }
res1: Int = 2

在您的情况下，以下内容基本上等同于您的代码：

val k = (array zip array.tail) lastIndexWhere { case (x, y) => x < y }

我正在使用lastIndexWhere而不是indexWhere，因为在你的代码中，当你遇到一个谓词所在的对时，你不会停止循环。

Answer 2

sliding为您提供了进入集合的滑动窗口，即

scala> Array(1,2,2,4,5,6, 6).sliding(2).toList
res12: List[Array[Int]] = List(Array(1, 2), Array(2, 2), Array(2, 4), Array(4, 5), Array(5, 6), Array(6, 6))

这样可以轻松找到第一个匹配对的索引：

Array(1,2,2,4,5,6, 6).sliding(2).indexWhere { case Array(x1, x2) => x1 == x2 }

只提供第一个索引，使用collect来捕捉所有索引！

Array(1,2,2,4,5,6, 6)
  .sliding(2)     //splits each in to pairs
  .zipWithIndex   //attaches the current index to each pair
  .collect { case (Array(x1, x2), index) if (x1 == x2) => index }  //collect filters out non-matching pairs AND transforms them to just the inde

用于在数组中查找索引的功能代码

2 个答案: