vector - 在Rust中，有没有一种方法可以仅在容器的一部分上执行retain()？

我正在学习Rust的乐趣，在编写返回第n个素数的函数时，我意识到我的解决方案不雅致且浪费。

当我只想遍历尚未验证的部分时，我遍历整个可能的素数。使用keep()函数时，有没有办法做到这一点？

我的主要问题是，我相信retain()函数是为Vec实现的，而不是为切片实现的。我尝试研究切片文档以获取进一步的想法，但找不到任何想法。

这是代码。

let mut prime_list: Vec<u32> = (2..n).collect();
let mut i: usize = 0;
while i < prime_list.len() {
    let prime: u32 = prime_list[i];
    prime_list.retain(
       |x| x <= &prime || x % &prime != 0);
    i += 1;
}

所以问题出在函数调用中

prime_list.retain(
    |x| x <= &prime || x % &prime != 0);

每次进入此循环时，我不需要遍历列表中的每个元素，因为我知道我已经在“i”之前验证了元素。

但是keep()函数需要遍历整个容器，因此我在其中添加了一个检查(x
我想做的就是优雅地从位置“i”的元素开始，然后遍历列表删除元素。

我知道我可以用不太优雅的while或for循环来做到这一点，但我希望有一种Rust的方式来优雅地做到这一点。

谢谢你的帮助!

编辑:这是一个可能的拆分和filter()解决方案，感觉比我想的还要困惑...如果有更好的方法使用filter()方法和迭代器，我将非常感谢右边的轻推方向。

// inside the while loop
let prime: u32 = prime_list[i];
let (left, right) = prime_list.split_at(i);

prime_list = left.iter()
    .chain(
         right.iter()
         .filter(|x| *x == &prime || *x % &prime != 0))
    .map(|&x| x)
    .collect();

i += 1;

最佳答案

这并不能完全解决您想要的问题，但是我想我会做一个更“使用rust ”的方法，作为一个已经写了一段时间 rust 迹的人。

let mut primes = Vec::new();

for e in 2..100 {
    if primes.iter().all(|p| e % p != 0) {
        primes.push(e);
    }
}

println!("{:?}", primes);

我将质数存储在一个单独的向量中，因为我想在添加时将其添加。与在迭代过程中尝试使用保留从向量中删除某些元素相比，这确实感觉更优雅。

如果您真的想花哨，可以使用with_capacity进行一些优化，并在获得第n个素数后停止。

fn get_nth_prime(n: usize) -> u32 {
    let mut primes = Vec::with_capacity(n);
    let mut e = 2;

    while primes.len() < n {
        if primes.iter().all(|p| e % p != 0) {
            primes.push(e);
        }

        e += 1;
    }

    primes[n - 1]
}

关于vector - 在Rust中，有没有一种方法可以仅在容器的一部分上执行retain()？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/56615602/