scala中的迭代器

34 阅读2分钟

一,迭代器的定义

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

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

package caseclass1

/*
*
*   迭代器
*
 */

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

    list1.foreach(ele => println(ele))

    //3,4

    val iterator = list1.iterator
    // hasNext
    // next
    while (iterator.hasNext){
      println(iterator.next())
    }
  }
}

二,迭代器的基本使用

我们来看实际的需求。遍历输出List中的元素

package caseclass1

/*
 * 迭代器
 * 1. next() 获取下一个,移动迭代器的位置。如果没有下一个,就报错
 * 2. hasNext判断当前是否存在下一个元素: false,true
 * 3. drop()
 *
 */
object class6 {
  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.hasNext) // 当前是否还有 下一个元素

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

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

第一步:构建一个迭代器

第二步:循环获取元素

三,迭代器的优点

讲解对比foreach它的优点在于:

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

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

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

object T1{

  def main(args: Array[String]): Unit = {

    // 创建一个字符串集合

    val set = Set("apple", "banana", "cherry")

    // 创建集合的迭代器

    val setIterator = set.iterator

    // 迭代器链式调用方法

    setIterator.map(s => "I like " + s).foreach(println)

  }
}

四,迭代器的常见方法

package caseclass1

/*
* 迭代器
* 1. next() 获取下一个,移动迭代器的位置。如果没有下一个,就报错
* 2. hasNext判断当前是否存在下一个元素: false,true
* 3. drop() 跳过指定数量的元素,返回一个新的迭代器
* 4. take() 获取指定数量的元素,返回一个新的迭代器
*/
object class7 {
  def main(args: Array[String]): Unit = {
  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
  }
  }
}