当前位置：首页 > article >正文

深入理解 Rust 的迭代器：从基础到高级

article 2025/2/21 9:07:45

1. 迭代器的基础概念

1.1 什么是迭代器？

迭代器是一种设计模式，允许我们逐个访问集合中的元素，而无需暴露集合的内部结构。在 Rust 中，迭代器通过实现 Iterator trait 来定义。该 trait 主要包含一个方法：

pub trait Iterator {
    type Item;

    fn next(&mut self) -> Option<Self::Item>;
}

Item：关联类型，表示迭代器返回的元素类型。
next：返回迭代器的下一个元素，类型为 Option<Self::Item>。当迭代器结束时，返回 None。

1.2 创建迭代器

Rust 的标准库为多种类型提供了迭代器方法。例如，Vec 类型提供了 iter 方法：

let v = vec![1, 2, 3];
let mut iter = v.iter();

这将创建一个对 v 的不可变引用的迭代器。需要注意的是，调用 iter 方法不会立即执行任何操作，迭代器是惰性求值的，只有在消费迭代器时才会开始工作。

2. 使用迭代器

2.1 使用 `for` 循环

Rust 的 for 循环与迭代器紧密结合。它会自动调用迭代器的 next 方法，直到迭代器返回 None：

let v = vec![1, 2, 3];
for val in v.iter() {
    println!("Got: {}", val);
}

这段代码会输出：

Got: 1
Got: 2
Got: 3

2.2 显式调用 `next` 方法

除了使用 for 循环外，您还可以显式地调用迭代器的 next 方法：

let v = vec![1, 2, 3];
let mut iter = v.iter();

while let Some(val) = iter.next() {
    println!("Got: {}", val);
}

这与前面的示例效果相同，但提供了更多的控制权。

3. 迭代器适配器

3.1 惰性求值

迭代器适配器是惰性求值的，这意味着它们不会立即执行操作，直到被消费。例如，map 方法返回一个新的迭代器，该迭代器在被消费时才会对每个元素应用指定的闭包。

let v = vec![1, 2, 3];
let v2: Vec<_> = v.iter().map(|x| x + 1).collect();
println!("{:?}", v2); // 输出: [2, 3, 4]

在这个例子中，map 创建了一个新的迭代器，该迭代器将每个元素加 1。只有在调用 collect 时，迭代器才会被消费，闭包才会被执行。

3.2 链式调用

您可以将多个迭代器适配器链式调用，以实现复杂的操作：

let v = vec![1, 2, 3, 4, 5];
let v2: Vec<_> = v.iter()
    .filter(|&x| x % 2 == 0)
    .map(|x| x * 2)
    .collect();
println!("{:?}", v2); // 输出: [4, 8]

这段代码首先过滤出偶数，然后将每个偶数乘以 2，最后收集结果。

4. 迭代器的消费

4.1 消费适配器

迭代器的消费适配器会消耗迭代器并返回一个值。例如，sum 方法会遍历迭代器的所有元素，将它们相加，并返回总和：

let v = vec![1, 2, 3];
let sum: i32 = v.iter().sum();
println!("Sum: {}", sum); // 输出: Sum: 6

4.2 终止适配器

终止适配器会在满足特定条件时停止迭代。例如，find 方法会返回第一个满足条件的元素：

let v = vec![1, 2, 3, 4, 5];
let first_even = v.iter().find(|&&x| x % 2 == 0);
println!("{:?}", first_even); // 输出: Some(2)

5. 自定义迭代器

您还可以为自定义类型实现 Iterator trait，以使其可迭代：

struct Counter {
    count: u32,
    max: u32,
}

impl Counter {
    fn new(max: u32) -> Self {
        Counter { count: 0, max }
    }
}

impl Iterator for Counter {
    type Item = u32;

    fn next(&mut self) -> Option<Self::Item> {
        if self.count < self.max {
            self.count += 1;
            Some(self.count)
        } else {
            None
        }
    }
}

let counter = Counter::new(5);
for num in counter {
    println!("{}", num);
}