根据某些条件从 Vec 中删除元素 [重复]

Question

我的代码如下所示：

struct Bar {
    i: i32,
}

struct Foo {
    v: Vec<Bar>,
}

impl Foo {
    fn foo(&mut self) {
        self.v.drain(self.v.iter().filter(|b| b.i < 10));
    }
}

注意Bar不允许被复制或克隆。

我想删除所有不满足该条件的Bars。最初我以为我可以遍历它们并调用remove，但如果有一个完全有意义的不可变借用，我不允许有两个可变借用或一个可变借用。

我想最简单的方法是只使用clone、filter 和collect，但我不允许克隆或复制。

我该怎么做？

Answer 1

在夜间 Rust 中，您可以使用 Vec::drain_filter:

#![feature(drain_filter)]

#[derive(Debug)]
struct Bar {
    i: i32,
}

fn main() {
    let mut bars = vec![Bar { i: 1 }, Bar { i: 10 }, Bar { i: 3 }, Bar { i: 100 }];
    bars.drain_filter(|b| b.i < 10);
    println!("{:?}", bars);
}

drain_filter 的额外有趣之处在于，当它返回它们的迭代器时，您可以获得被拒绝的值：

let rejects: Vec<_> = bars.drain_filter(|b| b.i < 10).collect();

您也可以选择修改被迭代的值：

bars.drain_filter(|b| {
    b.i -= 1;
    b.i < 10
});

Answer 2

如果你查看Vec 的接口，你不会发现基于谓词擦除某些元素的方法。相反，您会发现 retain 基于谓词保留元素。

当然，两者都是对称的，只是retain 更难找到如果您通过“删除”或“擦除”过滤方法名称（它的描述中确实包含“删除”）。

提供的示例不言自明：

let mut vec = vec![1, 2, 3, 4];
vec.retain(|&x| x % 2 == 0);
assert_eq!(vec, [2, 4]);