如何从迭代器循环内部跳过 n 个项目？

Question

这段代码：

play

fn main() {
    let text = "abcd";

    for char in text.chars() {
        if char == 'b' {
            // skip 2 chars
        }
        print!("{}", char);
    }
    // prints `abcd`, but I want `ad`
}

打印abcd，但如果找到b，我想跳过2个字符，以便打印ad。我怎么做？

我试图将迭代器放入循环外的变量中并在循环内操作该迭代器，但借用检查器不允许这样做。

Answer 1

AFAIK 你不能用for 循环来做到这一点。您需要手动脱糖：

let mut it = text.chars();
while let Some(char) = it.next() {
    if char == 'b' {
        it.nth(1); // nth(1) skips/consumes exactly 2 items
        continue;
    }
    print!("{}", char);
}

Playground

Answer 2

如果你想保持迭代器风格，你可以使用std::iter::successors（为了更易读，我已经用'!'替换了特殊字符：

fn my_iter<'a>(s: &'a str) -> impl Iterator<Item = char> + 'a {
    let mut it = s.chars();

    std::iter::successors(it.next(), move |c| {
        if *c == '!' {
            it.next().and_then(|_| it.next())
        } else {
            it.next()
        }
    })
    .filter(|c| *c != '!')
}

fn main() {
    assert!(my_iter("a!bc").eq("ac".chars()));
    assert!(my_iter("!abcd").eq("bcd".chars()));
    assert!(my_iter("abc!d").eq("abc".chars()));
    assert!(my_iter("abcd!").eq("abcd".chars()));
}