scala - 在数组中查找索引的功能代码

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