scala中的迭代器

60 阅读3分钟

一.迭代器的定义

迭代器是一种用于遍历集合元素的对象。它提供了统一的方式来访问各种集合类 型(如列表、映射、集合等)中的元素,而不需要了解集合的内部结构。

在 Scala 中,迭代器是一种抽象的概念,它有特定的接口和方法来实现元素的遍历。

package iterator

object iterator01 {
  def main(args: Array[String]): Unit = {
    val list = List(1, 2, 3, 4, 5)

    // 依次输出所有元素
    list.foreach(ele => print(ele))
    println() // 换行

    // 仅输出3、4
    val it = list.iterator.drop(2).take(2)
    while (it.hasNext) {
      println(it.next())
    }
  }
}

二.迭代器的基本使用

(1) iterator方法创建对应的迭代器。这个迭代器可以用于遍历列表中的元素。

(2) hasNext方法检查是否还有下一个元素。

(3) next方法获取一个元素

package iterator

object iterator02 {
  // 迭代器核心方法说明:
  // 1. next():获取当前元素,同时移动迭代器指针
  // 2. hasNext():判断迭代器是否还有下一个元素
  // 3. drop(n):从当前位置跳过n个元素,返回新的迭代器

  def main(args: Array[String]): Unit = {
    // 获取List(1,2,3,4)的迭代器
    val it = List(1, 2, 3, 4).iterator

    // 跳过前2个元素,生成新迭代器it1(此时it1指向3)
    val it1 = it.drop(2)

    // 遍历it1,输出剩余元素
    while (it1.hasNext) {
      println(it1.next()) // 输出:3、4
    }
  }
}

使用迭代器的基本步骤是:

第一步:构建一个迭代器

第二步:循环获取元素

三.迭代器的优点

(1)内存效率高。迭代器采用延迟计算的方式,它不会将整个集合加载到内存中, 而是在每次调用next方法时才计算并返回下一个元素。

(2)统一的遍历方法。迭代器为不同类型的集合(如列表、集合、映射等)提供了 一种统一的遍历方式。无论底层集合的具体结构如何,只要能获取到迭代器,就 可以使用相同的hasNext和next方法来遍历元素。

(3)支持函数式编程风格,便于链式操作。

四.迭代器的常见方法

  1. next方法: 作用:获取迭代器的下一个元素,并将迭代器的位置向前移动一位。如果没 有下一个元素,调用next方法会抛出 NoSuchElementException异常。
  2. 子迭代器(以 drop 和 take 为例)
  3. duplicate 方法
  4. toList 方法
  5. zip 方法
package iterator

object iterator03 {
  /**
   * 1. next() 获取当前的元素,移动一次迭代器
   * 2. hasNext 判断是否有下一个元素
   * 3. drop(n) 从当前位置开始,跳过 n 个元素,它返回一个新的迭代器
   * 4. take(n) 从当前位置开始,获取 n 个元素,它返回一个新的迭代器
   * 5. duplicate 复制迭代器
   * 6. toList 把迭代器中剩余的元素放在一个新的list中
   * 7. zip 把两个迭代器一对一的绑定起来,可以同步移动这两个迭代器,并且以元组的方式
   */
  def main(args: Array[String]): Unit = {
    val it = List(1,2,3,4,5).iterator

    it.next()
    it.next()
    val list1 = it.toList
    println(list1)

    val it1 = List(1,2,3,4,5,6).iterator
    val it2 = List("one", "two", "three", "four", "five").iterator

    val it3 = it1.zip(it2)

    while(it3.hasNext){
      val rs = it3.next()
      println(rs._1 + " " + rs._2)
    }
  }
}