Iterator和IntoIterator

  1. trait Iterator {
  2. type Item;
  3. fn next(&mut self) -> Option<Self::Item>;
  4. ... // many default methods
  5. }

迭代器都实现了Iterator,他的关联类型Item是迭代器输出的每个元素的类型。

如果一个类型有一个比较自然的变成迭代器的方法,那这个类型可以实现IntoIterator。

  1. trait IntoIterator where Self::IntoIter: Iterator<Item=Self::Item> {
  2. type Item;
  3. type IntoIter: Iterator;
  4. fn into_iter(self) -> Self::IntoIter;
  5. }

从定义可以看出,生成迭代器会获取本身的所有权。关联类型IntoIter是返回的迭代器,Item是这个迭代器元素的类型。实现了该trait的类型就是一个Iterable。

for循环就是调用了IntoIteratorIterator的方法。如果调用的值本身就是个Iterator也没问题,因为Iterator本身也实现了IntoIterator

创建迭代器

从集合创建

大多数集合类型都有iteriter_mut方法。分别返回不变引用和可变引用的迭代器。
因为有些类型没有自然的成为迭代器的方法,需要有单独的逻辑来创建适合的迭代器。比如&str,可以使用bytes方法转成字节按字节迭代,也可以用chars方法,转成字符向量按字符迭代。

for循环里除了可以用值本身以外也可以用引用和可变引用,迭代生成的元素也分别是原集合中元素的引用和可变引用。这是因为被迭代的集合类型的三个类型&T&mut TT本身都实现了IntoIterator。但不是所有的类型都是这样。 HashSet, BTreeSet, 和BinaryHeap没有实现可变引用的IntoIterator。因为改变的元素可能会影响整个集合的结构。HashMapBTreeMap可以生成可变的值但是不可变的键。

(&T).into_iter()T.inter()是等价的,后者在不是for这种必须使用trait的情况下更简洁。
IntoIterator用在泛型变成中更方便,
T: IntoIterator可以限制类型是能生成迭代器的,
T: IntoIterator<Item=U>规定生成的迭代的输出的元素的类型是U。

用函数创建

  1. use rand::random; // In Cargo.toml dependencies: rand = "0.7"
  2. use std::iter::from_fn;
  3. // Generate the lengths of 1000 random line segments whose endpoints
  4. // are uniformly distributed across the interval [0, 1]. (This isn't a
  5. // distribution you're going to find in the `rand_distr` crate, but
  6. // it's easy to make yourself.)
  7. let lengths: Vec<f64> =
  8. from_fn(|| Some((random::<f64>() - random::<f64>()).abs()))
  9. .take(1000)
  10. .collect();

from_fn的参数是一个返回Option的函数,如果返回的Some则继续迭代,如果None则停止,上例中的函数一直是Some,所以用take函数只去1000个。collect从结果生成一个Vec

successor函数可以利用前一个值来生成下一个值。他的参数是一个初始值,和一个函数,这个函数有一个参数是前一个值,返回是迭代的下一个值。

  1. use num::Complex;
  2. use std::iter::successors;
  3. fn escape_time(c: Complex<f64>, limit: usize) -> Option<usize> {
  4. let zero = Complex { re: 0.0, im: 0.0 };
  5. successors(Some(zero), |&z| { Some(z * z + c) })
  6. .take(limit)
  7. .enumerate()
  8. .find(|(_i, z)| z.norm_sqr() > 4.0)
  9. .map(|(i, _z)| i)
  10. }

find查找迭代中第一个出现的复合条件的值,返回Some(i, x)i是序号,x是值。

from_fnsuccessor参数中的函数都是FnMut

很多集合类型有drain方法,从值的可变引用生成迭代器,并消耗迭代器中涉及的元素,不管是不是用到了。

  1. use std::iter::FromIterator;
  2. let mut outer = "Earth".to_string();
  3. let inner = String::from_iter(outer.drain(1..4));
  4. assert_eq!(outer, "Eh");
  5. assert_eq!(inner, "art");

标准库中的其他迭代器

Type or trait Expression Notes
std::ops::Range iterator 1..10 Endpoints must be an integer type to be iterable. Range includes start value and excludes end value.
(1..10).step_by(2) Produces 1, 3, 5, 7, 9.
std::ops::RangeFrom 1.. Unbounded iteration. Start must be an integer. May panic or overflow if the value reaches the limit of the type.
std::ops::RangeInclusive 1..=10 Like Range, but includes end value.
Option Some(10).iter() Behaves like a vector whose length is either 0 (None) or 1 (Some(v)).
Result Ok(“blah”).iter() Similar to Option, producing Ok values.
Vec, &[T] v.windows(16) Produces every contiguous slice of the given length, from left to right. The windows overlap.
v.chunks(16) Produces nonoverlapping, contiguous slices of the given length, from left to right.
v.chunks_mut(1024) Like chunks, but slices are mutable.
v.split(|byte| byte & 1 != 0) Produces slices separated by elements that match the given predicate.
v.split_mut(…) As above, but produces mutable slices.
v.rsplit(…) Like split, but produces slices from right to left.
v.splitn(n, …) Like split, but produces at most n slices.
String, &str s.bytes() Produces the bytes of the UTF-8 form.
s.chars() Produces the chars the UTF-8 represents.
s.split_whitespace() Splits string by whitespace, and produces slices of nonspace characters.
s.lines() Produces slices of the lines of the string.
s.split(‘/‘) Splits string on a given pattern, producing the slices between matches. Patterns can be many things: characters, strings, closures.
s.matches(char::is_numeric) Produces slices matching the given pattern.
std::collections::HashMap,
std::collections::BTreeMap
map.keys(),
map.values()
Produces shared references to keys or values of the map.
map.values_mut() Produces mutable references to entries’ values.
std::collections::HashSet,
std::collections::BTreeSet
set1.union(set2) Produces shared references to elements of union of set1 and set2.
set1.intersection(set2) Produces shared references to elements of intersection of set1 and set2.
std::sync::mpsc::Receiver recv.iter() Produces values sent from another thread on the corresponding Sender.
std::io::Read stream.bytes() Produces bytes from an I/O stream.
stream.chars() Parses stream as UTF-8 and produces chars.
std::io::BufRead bufstream.lines() Parses stream as UTF-8, produces lines as Strings.
bufstream.split(0) Splits stream on given byte, produces inter-byte Vec buffers.
std::fs::ReadDir std::fs::read_dir(path) Produces directory entries.
std::net::TcpListener listener.incoming() Produces incoming network connections.
Free functions std::iter::empty() Returns None immediately.
std::iter::once(5) Produces the given value and then ends.
std::iter::repeat(“#9”) Produces the given value forever.

迭代适配器

适配器都是从一个迭代器产生另一个迭代器。调用适配器的时候还没有开始迭代,只有在最后一个适配器调用next的时候才开始。也就是惰性的。
Rust的迭代适配器是没有overhead的,会用inline来让整个代码变成一个整体。就像是手写的循环一样。

map和filter

map从获取迭代的元素,产生一个新的。
filter获取的是引用。

flatten可以用来过滤Some和Ok

peek

就是让你用下一个元素,但不消耗它。

fuse

虽然大多数迭代器穷尽之后会返回None,但不是所有的都这样,所以不能依赖这个行为

zip

  1. use std::iter::repeat;
  2. let endings = vec!["once", "twice", "chicken soup with rice"];
  3. let rhyme: Vec<_> = repeat("going")
  4. .zip(endings)
  5. .collect();
  6. assert_eq!(rhyme, vec![("going", "once"),
  7. ("going", "twice"),
  8. ("going", "chicken soup with rice")]);

by_ref

一般的适配器会获取迭代器的所有权,执行结束之后,迭代器也被耗尽了。by_ref可以生成一个迭代器的可变引用给适配器,这样适配器在运行到某个地方停止后,原迭代器还是存在,迭代位置停留在了之前停止的地方。

cloned和copied

从元素是引用的迭代器,复制一个元素是值的迭代器。

cycle

将一个迭代器不断循环

消耗迭代器

count, sum, product

max, min

不能用于浮点数,因为浮点数只实现了PartialOrd。

max_by, min_by

用自定义的闭包确定原色的大小,来取最大或最小值,可以解决max,min不能比较浮点数的问题。
闭包的函数前面里,参数使用的是元素的引用。

max_by_key, min_by_key

用闭包生成一个可以比较值,用这个值来取结果

position, rposition, and ExactSizeIterator

position查找某个元素的位置,rposition也是,但是从右边开始找,因此rposition只有reversible的迭代器才能用,而且这个迭代器也需要实现:

  1. trait ExactSizeIterator: Iterator {
  2. fn len(&self) -> usize { ... }
  3. fn is_empty(&self) -> bool { ... }
  4. }

len方法返回迭代器还剩下的原色个数。不是所有的迭代器都能确定剩下的个数,比如一些无限的迭代器,又比如str需要解析后面的元素那些字节是一个utf-8才能返回一个元素。

fold, rfold, try_fold, try_rfold

fold 就是reduce。
try版本的返回Result或Option,如果结果是Err或None,迭代停止,且不会消耗迭代器。

nth, nth_back

相当于一次执行n个next

last

消耗整个迭代器,去最后一个元素,即使是reversible的。

collect和size_hint

collect需要对应的类型实现FromIterator。一般的实现都是创建一个该类型的对象然后把迭代的值一个一个放进去,但这对于需要动态申请内存的类型(比如Vec)性能很差。因为有可能事先知道结果所需要的内存,所以Iterator有个方法size_hint,会返回预计该类型的size的范围。

实现迭代器

一个迭代一般都需要一个记录迭代状态的数据结构,像range这样的类型,start值可以记录状态,他的into_iter方法甚至可以用自己本身来实现,每迭代一个就加1。像二叉树就需要复杂一点的结构,需要实现IntoIterator,用into_iter方法生成一个专门的Iterator。书中的例子用树的左视图来记录未访问到 节点。

  1. impl<'a, T: 'a> TreeIter<'a, T> {
  2. fn push_left_edge(&mut self, mut tree: &'a BinaryTree<T>) {
  3. while let NonEmpty(ref node) = *tree {
  4. self.unvisited.push(node);
  5. tree = &node.left;
  6. }
  7. }
  8. }

便利的时候更新该左视图

  1. impl<'a, T> Iterator for TreeIter<'a, T> {
  2. type Item = &'a T;
  3. fn next(&mut self) -> Option<&'a T> {
  4. // Find the node this iteration must produce,
  5. // or finish the iteration. (Use the `?` operator
  6. // to return immediately if it's `None`.)
  7. let node = self.unvisited.pop()?;
  8. // After `node`, the next thing we produce must be the leftmost
  9. // child in `node`'s right subtree, so push the path from here
  10. // down. Our helper method turns out to be just what we need.
  11. self.push_left_edge(&node.right);
  12. // Produce a reference to this node's value.
  13. Some(&node.element)
  14. }
  15. }