scala中的迭代器

33 阅读2分钟

迭代器的定义

迭代器是一种用于遍历集合元素的对象。它提供了统一的方式来访问各种集合类 型(如列表、映射、集合等)中的元素,而不需要了解集合的内部结构。 在 Scala 中,迭代器是一种抽象的概念,它有特定的接口和方法来实现元素的遍历。

迭代器的基本使用

// 创建一个整数列表
val list = List(1, 2, 3, 4, 5)
// 通过 iterator 方法创建对应的迭代器
val iterator = list.iterator  
 while (iterator.hasNext) {
   println(iterator.next())

}

使用迭代器步骤:

  1. iterator方法创建对应的迭代器。这个迭代器可以用于遍历列表中的元素。
  2. hasNext方法检查是否还有下一个元素。
  3. next方法获取一个元素

迭代器的优点

  1. 内存效率高。迭代器采用延迟计算的方式,它不会将整个集合加载到内存中,而是在每次调用next方法时才计算并返回下一个元素。
  2. 统一的遍历方法。迭代器为不同类型的集合(如列表、集合、映射等)提供了 一种统一的遍历方式。无论底层集合的具体结构如何,只要能获取到迭代器,就 可以使用相同的hasNextnext方法来遍历元素。
  3. 支持函数式编程风格,便于链式操作。

迭代器的常见方法

  1. next()获取下一个,移动迭代器的位置,如果没有下一个,就报错
  2. hasNext()判断当前是否存在下一个元素:false,true
  3. drop()跳过指定数量的元素,返回一个新的迭代器
  4. take()获取指定数量的元素,返回一个新的迭代器

完整代码示例:

package list
/*
   1.next()获取下一个,移动迭代器的位置,如果没有下一个,就报错
   2.hasNext()判断当前是否存在下一个元素:false,true
   3.drop()跳过指定数量的元素,返回一个新的迭代器
   4.take()获取指定数量的元素,返回一个新的迭代器
 */
object class05 {
  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)
//
//    println(it.next())
     val it1=List(1,2,3,4,5).iterator
    it1.next()
    it1.next()
    //跳过前两个元素
    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
    }
  }
}