迭代器元素的条件拼接——Scala惯用解决方案

嘿，你说得太对了！确实有更贴合Scala函数式编程风格的写法——我们可以直接利用Scala标准库中的惰性迭代器操作，不用手动定义新的Iterator子类，同时还能严格满足「每次内存中仅保留单个分组数据」的要求。

方案一：使用Iterator.unfold（Scala 2.13+推荐） #

unfold是Scala 2.13引入的惰性迭代器生成工具，它会帮我们维护迭代器的剩余状态，每次只生成一个分组的结果，完全不会占用额外内存存储所有分组：

object IteratingAndAccumulating extends App {
  val inputLines = Iterator("a", "b", "c break", "d break", "e")

  // 惰性生成每个分组的拼接字符串
  val groupedLines: Iterator[String] = Iterator.unfold(inputLines) { currentIt =>
    if (currentIt.isEmpty) None // 迭代器为空时停止
    else {
      // 先取所有不满足"break结尾"的元素
      val (nonBreakLines, rest) = currentIt.span(line => !line.endsWith("break"))
      // 如果剩余迭代器还有元素，第一个就是带break的，要加入当前分组
      val fullGroup = if (rest.hasNext) nonBreakLines ++ Iterator(rest.next()) else nonBreakLines
      // 返回当前分组的拼接结果，以及剩余的迭代器状态
      Some(fullGroup.mkString("-"), rest)
    }
  }

  // 输出结果
  groupedLines.zipWithIndex.foreach { case (concatenated, idx) =>
    println(s"(Group: $idx): $concatenated")
  }
}

方案二：函数式递归实现 #

如果你需要兼容更早的Scala版本，也可以用递归的方式实现，全程避免可变容器（比如原代码中的ListBuffer）：

object IteratingAndAccumulating extends App {
  val inputLines = Iterator("a", "b", "c break", "d break", "e")

  def groupByBreak(it: Iterator[String]): Iterator[String] = {
    if (!it.hasNext) Iterator.empty
    else {
      val firstLine = it.next()
      // 构建当前分组：如果首行是break结尾，直接作为分组；否则取所有非break行，再加上下一个break行（如果存在）
      val currentGroup = 
        if (firstLine.endsWith("break")) Iterator(firstLine)
        else Iterator(firstLine) ++ it.takeWhile(!_.endsWith("break")) ++ (if (it.hasNext) Iterator(it.next()) else Iterator.empty)
      // 递归处理剩余迭代器，拼接结果迭代器
      Iterator(currentGroup.mkString("-")) ++ groupByBreak(it)
    }
  }

  // 输出结果
  groupByBreak(inputLines).zipWithIndex.foreach { case (concatenated, idx) =>
    println(s"(Group: $idx): $concatenated")
  }
}

为什么这两种写法更Scala？ #

• 无可变状态：避免了原代码中ListBuffer这种可变容器，完全遵循函数式编程的不可变原则；
• 惰性计算：所有操作都是惰性的，只会在需要的时候生成下一个分组，内存中始终只保留当前处理的分组数据；
• 复用标准库：不用手动实现Iterator的hasNext和next方法，利用Scala内置的span、takeWhile、unfold等方法，代码更简洁易读，也更少出错。

内容的提问来源于stack exchange，提问作者Fil Karnicki