假设我有一个向量,它上面有一个过滤器,我想把它分解成两个向量:一个过滤向量和它的补码.我认为有两种方法可以做到这一点.第一个是for循环:

let vct: Vec<u32> = vec![1, 3, 4, 7, 9, 10, 12];
let filter = |x| x % 3 == 0;

let mut filtered: Vec<u32> = Vec::new();
let mut complement: Vec<u32> = Vec::new();
for v in vct {
    if filter(v) {
        filtered.push(v);
    } else {
        complement.push(v);
    }
}

println!("{:?}", filtered);   //[3, 9, 12]
println!("{:?}", complement); //[1, 4, 7, 10]

这似乎很有效,但不必要地冗长(并为向量添加了不必要的可变性).另一种是使用迭代器的简单方法:

let vct: Vec<u32> = vec![1, 3, 4, 7, 9, 10, 12];
let filter = |x:&&u32| *x % 3 == 0;

let filtered: Vec<u32> = vct.iter().filter(filter).cloned().collect();
let complement: Vec<u32> = vct.into_iter().filter(|x| !filter(&x)).collect();

println!("{:?}", filtered);   //[3, 9, 12]
println!("{:?}", complement); //[1, 4, 7, 10]

它有一个更清晰的意图,也不太冗长(除了filter定义/用法,这有点难看),但在数组上迭代twice次,这似乎是不必要的.

这两种解决方案似乎都不是最优的.

是否有一种惯用的and有效方法(无需重新迭代)将向量拆分为过滤器及其补码?

推荐答案

您try 执行的操作称为partitioning.您可以使用Iterator.partition():

let vct: Vec<u32> = vec![1, 3, 4, 7, 9, 10, 12];
let filter: fn(u32) -> bool = |x| x % 3 == 0;

let (filtered, complement): (Vec<u32>, Vec<u32>) = vct.iter().partition(|x| filter(**x));

assert_eq!(&[3, 9, 12], filtered.as_slice());
assert_eq!(&[1, 4, 7, 10], complement.as_slice());

Playground

Rust相关问答推荐

将已知大小的切片合并成一个数组,

为什么要在WASM库中查看Rust函数需要`#[no_mangle]`?

在Rust中赋值变量有运行时开销吗?

将大小为零的类型实例存储到空指针中

如何获取Serde struct 的默认实例

有没有办法避免在While循环中多次borrow `*分支`

AXUM一路由多个不包括URL的参数类型

在文件链实施中绕过borrow 判断器

习语选项<;T>;到选项<;U>;当T->;U用From定义

由于生存期原因,返回引用的闭包未编译

Rust 中的静态引用

如何在 Rust 中将函数项变成函数指针

存储返回 impl Trait 作为特征对象的函数

Rust 中指向自身的引用如何工作?

为什么 Rust 字符串没有短字符串优化 (SSO)?

如何刷新 TcpStream

我什么时候应该使用特征作为 Rust 的类型?

为什么我不能克隆可克隆构造函数的Vec?

字符串切片的向量超出范围但原始字符串仍然存在,为什么判断器说有错误?

Cargo:如何将整个目录或文件包含在功能标志中?