小白学苑

Scala集合：Stream流

什么是Stream流?

根据Scala文档，流是类似于列表的数据结构，只是流的元素是惰性计算机制。因此，可以拥有无限长的流!

Stream（流）与List类似，但是它是延迟计算的，所以可以非常非常长。

// 创建
val stream = 1 #:: 2 #:: 3 #:: Stream.empty

// very very long
val steam = (1 to 100000000).toStream
steam.head
steam.tail

示例

【示例】学习如何使用Scala的不可变流来执行一些常见的操作，比如初始化一个无限流，使用cons操作符，从流中获取元素以及创建一个空流。

import scala.collection.immutable.Stream.cons

object StreamDemo {
  def main(args: Array[String]): Unit = {
    println("1: 使用 #:: 创建一个带有3个数字的流")
    // val stream1: Stream[Any] = 1 #:: 2 #:: 3 #:: Stream.empty
    val stream1 = 1 #:: 2 #:: 3 #:: Stream.empty

    // 当我们试图打印Stream的元素时，只打印了Stream的第一个元素!
    // 如前所述，Streams的元素是延迟计算的。
    println(s"stream1的元素 = $stream1")

    println("\n2: 使用Stream.cons 创建一个带有3个数字的流")
    // Stream.cons通常被称为Cons操作符
    val stream2: Stream[Int] = cons(1, cons(2, cons(3, Stream.empty) ) )
    // 只打印Stream的第一个元素
    println(s"stream2的元素 = $stream2")

    println("\n3: 使用take 函数打印来自stream2中的3个数字")
    print("从stream2中取前3个数字 = ")
    stream2.take(3).print

    print("\n从stream2中取前10个数字 = ")
    // 当试图从Stream中获取10个数字时，尽管它只包含3个元素，也不会抛出任何IndexOutOfBoundsException
    stream2.take(10).print

    println("\n\n4: 使用Stream.cons定义一个无限数字流")
    def inifiniteNumberStream(number: Int): Stream[Int] = Stream.cons(number, inifiniteNumberStream(number + 1))
    print("从无限数字流中只取前20个数字 = ")
    inifiniteNumberStream(1).take(20).print

    println("\n\n5: 使用Stream.from定义一个无限数字流")
    val stream3: Stream[Int] = Stream.from(1)
    print("从无限数字流中只取前20个数字 = ")
    stream3.take(20).print

    println("\n\n6: 初始化一个空的Stream")
    val emptyStream: Stream[Int] = Stream.empty[Int]
    println(s"Empty Stream = $emptyStream")
  }
}