如何组合对列表的子集并返回最后一个元素的索引

Question

假设我有一个按第一个元素排序的对列表，看起来像这样：

List((0.0,0.001), (0.001,1.0), (1.0,2.0), (2.0,3.0), (3.0,4.0), (4.0,5.0), (5.0,15.0), (15.0,30.0), (30.0,45.0), (45.0,60.0), (60.0,71.0))

这本质上是一个区间列表。

给定一个起始元素，例如对应于(2.0,3.0) 的元素3，以及一个区间长度，例如40，如果我从索引开始，我想找到40 结束的区间索引3.

所以本质上，对于从（在这种情况下）元素 3 开始的每个元素，我需要执行 _._2 - _._1 并将它们相加，直到长度超过 40。一旦我到达它，我需要停止。我不想map整个列表到持续时间（列表可能很长），无论如何我不想foldLeft所有这些，我想在达到所需的持续时间后立即停止.它类似于传统命令式for 循环的break。

如果您要使用普通的旧 Java 命令式地执行此操作，这将非常简单。有没有什么聪明的技巧可以在 Scala 中以函数式的方式做到这一点？

Answer 1

听起来scanLeft 正是您要查找的内容——它可以让您计算运行总数：

def findIt(intervals: Seq[(Double, Double)], value: Double, start: Int): Int =
  start + intervals.drop(start).scanLeft(0.0) {
    case (acc, (b, e)) => acc + e - b
  }.tail.indexWhere(_ > value)

或者，可能更清楚：

def intervalLength(interval: (Double, Double)) = interval._2 - interval._1

def findIt(intervals: Seq[(Double, Double)], value: Double, start: Int): Int =
  start + intervals
            .drop(start)
            .map(intervalLength)
            .scanLeft(0.0)(_ + _).tail.indexWhere(_ > value)

在您的情况下，此方法将计算整个列表的总数，但您可以通过将输入转换为 Stream 轻松使其变得懒惰。为了证明这是可行的，我们可以编写以下内容，其中xs 是您的示例列表：

scala> findIt(Stream.continually(xs).flatten, 40.0, 3)
res0: Int = 8

我们刚刚在无限流上运行findIt，这是因为永远重复您的数据，我们会立即得到结果，因此我们绝对不会计算所有总数。

Answer 2

def indexOf(intervals: List[(Double, Double)], start: Int, interval: Double): Option[Int] = {
  def recur(el: List[(Double, Double)], pos: Int, remaining: Double): Option[Int] = {
    def duration(p: (Double, Double)): Double = p._2 - p._1
    el match {
      case head :: _ if remaining - duration(head) < 0 => Some(pos)
      case head :: _ => recur(el.tail, pos+1, remaining - duration(head))
      case Nil => None
    }
  }
  recur(intervals.drop(start), start, interval)
}

示例用法：

val intervals = List((0.0,0.001), (0.001,1.0), (1.0,2.0), (2.0,3.0), (3.0,4.0), (4.0,5.0), (5.0,15.0), (15.0,30.0), (30.0,45.0), (45.0,60.0), (60.0,71.0))
println(indexOf(intervals, 0, 0.0005))
println(indexOf(intervals, 0, 0.001))
println(indexOf(intervals, 3, 40d))
println(indexOf(intervals, 3, 200d))

输出：

Some(0)
Some(1)
Some(8)
None