迭代列表，返回当前元素、下一个元素以及当前元素之前的元素

Question

我在以 scala 式且优雅的方式编写特定应用程序时遇到问题。我已经尝试了一段时间了，但我找不到解决此问题的“好的”解决方案：

鉴于我有以下列表：

List("foo", "bar", "baz", "blah")

我想迭代此列表，不仅为我提供每次迭代的当前元素，还为我提供当前元素之前和之后的元素。这可能是一个 Tuple3，但不是必需的。这可能是元组签名：

(Option[T], T, Option[T])

为了澄清我的意思，这是在 List[String] 上每次迭代的建议元组，在第四次之后结束。

迭代 1：(None, "foo", Some("bar"))

迭代 2：(Some("foo")，"bar"，Some("baz"))< /code>

迭代 3: (Some("bar"), "baz", Some("blah"))

迭代 4: (Some("baz"), "blah",无）

我怎样才能达到这样的结果？再次强调：我不受 Tuple3 的约束，任何其他解决方案也非常感谢！

谢谢！

Answer 1

这是一种方法。它使用新的 Scala 2.8 集合方法sliding。

def window[A](l: List[A]): Iterator[List[Option[A]]] = 
   (None :: l.map(Some(_)) ::: List(None)) sliding 3

window(List(1, 2, 3, 4, 5)).toList

// List(List(None, Some(1), Some(2)), List(Some(1), Some(2), Some(3)), List(Some(2), Some(3), Some(4)), List(Some(3), Some(4), Some(5)), List(Some(4), Some(5), None))

更新：这是适用于 Streams 的版本。

def windowS[A](s: Stream[A]): Stream[List[Option[A]]] = 
  (None #:: s.map(Some(_): Option[A]) #::: Stream(None: Option[A])).sliding(3).toStream.map(_.toList)  

val posInts = Stream.range(1, Integer.MAX_VALUE)
windowS(posInts).take(5).toList

Answer 2

如果您使用的是 2.8，Retronym 的答案效果很好。如果您使用的是 2.7.x，则没有很好的库存解决方案，但您可以轻松构建自己的解决方案。例如，如果您只想存在之前和之后的三元组，则可以执行以下操作：

class Tuple3Iterator[T](solo: Iterator[T]) extends Iterator[(T,T,T)] {
  var current = if (solo.hasNext) Some(solo.next) else None
  var future = if (solo.hasNext) Some(solo.next) else None
  def hasNext = solo.hasNext
  def next = {
    val past = current
    current = future
    future = Some(solo.next)
    (past.get,current.get,future.get)
  }
}
class IteratorToT3[T](it: Iterator[T]) {
  def treble = new Tuple3Iterator[T](it)
}
implicit def allowTrebling[T](it: Iterable[T]) = new IteratorToT3[T](it.elements)

scala> List("Hi","there",5,"you").treble.foreach(println(_))         
(Hi,there,5)
(there,5,you)

如果您更愿意允许之前和之后保留选项，（编辑：我并没有真正给出完整或无错误的一组之前更改）然后改为使用

class Tuple3Iterator[T](solo: Iterator[T]) extends Iterator[(Option[T],T,Option[T])] {
  var current = None:Option[T]
  var future = if (solo.hasNext) Some(solo.next) else None
  def hasNext = (solo.hasNext || future!=None)
  def next = {
    val past = current
    current = future
    future = if (solo.hasNext) Some(solo.next) else None
    (past,current.get,future)
  }
}

scala> List("Hi","there",5,"you").treble.foreach(println(_))
(None,Hi,Some(there))
(Some(Hi),there,Some(5))
(Some(there),5,Some(you))
(Some(5),you,None)

Answer 3

更好地使用 Scala 2.8 和 retronym 解决方案，但这是我针对 Scala 2.7 的解决方案：

class MyIterator[T](l: List[T]) extends Iterator[(Option[T],T,Option[T])] {
  var last: Option[T] = None
  var curr = l
  def hasNext = !curr.isEmpty
  def next = {
    val t = curr match {
      case first :: second :: tail => (last, first, Some(second))
      case first :: Nil => (last, first, None)
      case Nil => throw new java.util.NoSuchElementException
    }
    last = Some(curr.head)
    curr = curr.tail
    t
  }
}