scala - 如何在流中找到两个连续且相同的值？

Question

如何在 a 中找到两个连续且相同的值Stream并返回此“重复值”：

def succ: Stream[Int] => Int = str => ...

例如Stream(1, 2, 4, 3, 5, 5, 2, 2)会导致5.

如何做到这一点？

Answer 1

您可以混合使用Stream.slidingand Stream.collectFirst：

def succ(stream: Stream[Int]): Option[Int] =
  stream.sliding(2).collectFirst { case Stream(x, y) if x == y => x }

产生：

Stream(1, 2, 4, 3, 5, 5, 2, 2)
  .sliding(2) // Stream(Stream(1, 2), Stream(2, 4), Stream(4, 3), ...
  .collectFirst { case Stream(x, y) if x == y => x } // Some(5)

或者None如果没有连续的元素共享相同的值。

---

根据您的评论，为了返回0而不是None在没有连续元素共享相同值时返回，您可以.getOrElse(0)在管道末尾应用：

def succ(stream: Stream[Int]): Int =
  stream.sliding(2).collectFirst { case Stream(x, y) if x == y => x }.getOrElse(0)

Answer 2

用尾部压缩流是另一种处理单元素流的方法。空流被显式处理：

def succ: Stream[Int] => Int = {
  case Stream() => 0
  case str =>
    str.zip(str.tail).
    find { case (l, r) => l == r }.
    map { case (x, _) => x }.
    getOrElse(0)
}

Answer 3

使用简单递归怎么样？

def succ: Stream[Int] => Int = {
  def rec(x: Int, xs: Stream[Int]): Int = {
    if (x == xs.head) x
    else rec(xs.head, xs.tail)
  }

  (str: Stream[Int]) => rec(str.head, str.tail)
}

如果没有找到后续重复项，这将不起作用，因为您必须将返回类型更改为Option[Int]并添加一些额外的检查。