如何在过滤器闭包中使用Peekable迭代器?

时间:2017-12-05 01:05:09

标签: iterator rust

我只需找到下一个数字相同的数字:[1,2,2,3,4,4]应生成[2,4]。由于我需要查看下一个数字,我想我会尝试使用Peekable iterator并写一个filter

fn main() {
    let xs = [1, 2, 2, 3, 4, 4];
    let mut iter = xs.iter().peekable();

    let pairs = iter.filter(move |num| {
        match iter.peek() {
            Some(next) => num == next,
            None       => false,
        }
    });

    for num in pairs {
        println!("{}", num);
    }
}

我收到错误:

error[E0382]: capture of moved value: `iter`
 --> src/main.rs:6:15
  |
5 |     let pairs = iter.filter(move |num| {
  |                 ---- value moved here
6 |         match iter.peek() {
  |               ^^^^ value captured here after move
  |
  = note: move occurs because `iter` has type `std::iter::Peekable<std::slice::Iter<'_, i32>>`, which does not implement the `Copy` trait

我认为这是因为闭包使用iter,但它没有借用它,也无法复制它。

如何解决这个想要在过滤器中引用迭代器的问题?

1 个答案:

答案 0 :(得分:4)

  

引用过滤器内的迭代器

我不相信你可以。当你调用filter时,它取得了基础迭代器的所有权:

fn filter<P>(self, predicate: P) -> Filter<Self, P> 
where
    P: FnMut(&Self::Item) -> bool, 

一旦你这样做,它就消失了。不再有iter。在某些类似的情况下,您可以使用Iterator::by_ref可变地借用迭代器,将其驱动一段时间,然后再返回原始文件。在这种情况下,这不会起作用,因为内部迭代器需要第二次可变地借用它,这是不允许的。

  

仅查找下一个数字相同的数字。

extern crate itertools;

use itertools::Itertools;

fn main() {
    let input = [1, 2, 2, 3, 4, 4];

    let pairs = input
        .iter()
        .tuple_windows()
        .filter_map(|(a, b)| if a == b { Some(a) } else { None });

    let result: Vec<_> = pairs.cloned().collect();
    assert_eq!(result, [2, 4]);
}

或者如果你想要只使用标准库的东西:

fn main() {
    let xs = [1, 2, 2, 3, 4, 4];

    let mut prev = None;
    let pairs = xs.iter().filter_map(move |curr| {
        let next = if prev == Some(curr) { Some(curr) } else { None };
        prev = Some(curr);
        next
    });

    let result: Vec<_> = pairs.cloned().collect();
    assert_eq!(result, [2, 4]);
}