Question

我希望这个问题可以请函数编程爱好者。我是否可以要求一种方法将以下代码片段转换为Scala中的纯函数实现，并在可读性和执行速度之间取得良好的平衡？

描述：对于序列中的每个元素，生成一个子序列，其中包含当前元素（包括其自身）之后的元素，其距离小于给定阈值。超过阈值后，无需处理剩余的元素

  def getGroupsOfElements(input : Seq[Element]) : Seq[Seq[Element]] = {
    val maxDistance = 10 // put any number you may
    var outerSequence = Seq.empty[Seq[Element]]

    for (index <- 0 until input.length) {
      var anotherIndex = index + 1
      var distance = input(index) - input(anotherIndex) // let assume a separate function for computing the distance
      var innerSequence = Seq(input(index))

      while (distance < maxDistance && anotherIndex < (input.length - 1)) {
        innerSequence = innerSequence ++ Seq(input(anotherIndex))
        anotherIndex = anotherIndex + 1
        distance = input(index) - input(anotherIndex)
      }

      outerSequence = outerSequence ++ Seq(innerSequence)
    }

    outerSequence 
  }

Answer 1

您知道，如果您添加了对您尝试完成的代码以及代码的描述，这将会更容易。

无论如何，这里的东西可能接近你想要的东西。

def getGroupsOfElements(input: Seq[Element]): Seq[Seq[Element]] =
  input.tails.map(x => x.takeWhile(y => distance(x.head,y) < maxDistance)).toSeq

典型嵌套while循环的函数版本

1 个答案: