Scala中的迭代器

33 阅读2分钟

(一)迭代器的定义

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

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

1.next()获取下一个 移动迭代器的位置。如果没有下一个,就报错

2.hasNext判断当前是否存在下一个元素:false,true

object it01 {
  def main(args: Array[String]): Unit = {
    val it = List(1,2,3,4,5).iterator
    println(it.next())
    println(it.next())
    println(it.next())
    println(it.next())
    println(it.next())

    println(it.hasNext)//当前是否还有下一个元素

    //第六次使用next 报错!!
    println(it.next())
  }

}

运行结果:

1
2
3
4
5
false
Exception in thread "main" java.util.NoSuchElementException: head of empty list
	at scala.collection.immutable.Nil$.head(List.scala:652)
	at scala.collection.immutable.Nil$.head(List.scala:651)
	at scala.collection.StrictOptimizedLinearSeqOps$$anon$1.next(LinearSeq.scala:267)
	at it.it01$.main(it01.scala:19)
	at it.it01.main(it01.scala)

(二)迭代器的基本使用

遍历输出List中的元素。

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

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

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

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

第一步:构建一个迭代器

第二步:循环获取元素

(三)迭代器的优点

【对比foreach】它的优点在于:

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

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

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

(四)迭代器的常见方法

1. next方法

作用:获取迭代器的下一个元素,并将迭代器的位置向前移动一位。如果没 有下一个元素,调用next方法会抛出 NoSuchElementException异常。

2.hasNext判断当前是否存在下一个元素:false,true

3. 子迭代器(以 drop 和 take 为例)

  • drop()跳过指定数量的元素,返回一个新的迭代器

  • take()获取指定数量的元素,返回一个新的迭代器

object it01 {
  def main(args: Array[String]): Unit = {
//    val it = List(1,2,3,4,5).iterator
//    println(it.next())
//    println(it.next())
//    println(it.next())
//    println(it.next())
//    println(it.next())
//
//    println(it.hasNext)//当前是否还有下一个元素
//
//    //第六次使用next 报错!!
//    println(it.next())

    val it1 = List(1,2,3,4,5).iterator
    //跳过前两个元素
    val it2 = it1.drop(2)
    while(it2.hasNext){
      println(it2.next()) //3 4 5
    }

    val it3 = List(1,2,3,4,5).iterator
    it3.next()
    it3.next()
    val it4 = it3.take(2) //获取指定数量的元素,返回一个新的迭代器
    while(it4.hasNext){
      println(it4.next())//3 4
    }
  }

}