如何保留矢量元素及其原始索引?

Question

如果我有Vec，我可以通过v.iter().enumerate()使用索引迭代元素，并且我可以通过v.retain()删除元素。有没有办法同时做到这两点？

在这种情况下，索引不能再用于访问元素 - 它将是循环启动之前元素的索引。

我可以自己实现这一点，但为了尽可能高效.retain()我需要使用unsafe，我想避免这种情况。

这是我想要的结果：

let mut v: Vec<i32> = vec![1, 2, 3, 4, 5, 4, 7, 8];
v.iter()
.retain_with_index(|(index, item)| (index % 2 == 0) || item == 4);
assert(v == vec![1, 3, 4, 5, 4, 7]);

Answer 1

@Timmmm和@Hauleth的答案非常务实，我想提供几个替代方案。

这是一个带有一些基准和测试的游乐场： https://play.rust-lang.org/?version=nightly&mode=debug&edition=2018&gist=cffc3c39c4b33d981a1a034f3a092e7b

这很丑陋，但如果你真的想要一个v.retain_with_index()的方法，你可以对retain方法进行一些复制粘贴，并带有一个新的特征：

trait IndexedRetain<T> {
fn retain_with_index<F>(&mut self, f: F)
where
F: FnMut(usize, &T) -> bool;
}
impl<T> IndexedRetain<T> for Vec<T> {
fn retain_with_index<F>(&mut self, mut f: F)
where
F: FnMut(usize, &T) -> bool, // the signature of the callback changes
{
let len = self.len();
let mut del = 0;
{
let v = &mut **self;
for i in 0..len {
// only implementation change here
if !f(i, &v[i]) {
del += 1;
} else if del > 0 {
v.swap(i - del, i);
}
}
}
if del > 0 {
self.truncate(len - del);
}
}
}

这样，示例将如下所示：

v.retain_with_index(|index, item| (index % 2 == 0) || item == 4);

或者更好的是，你可以使用高阶函数：

fn with_index<T, F>(mut f: F) -> impl FnMut(&T) -> bool
where
F: FnMut(usize, &T) -> bool,
{
let mut i = 0;
move |item| (f(i, item), i += 1).0
}

这样示例现在如下所示：

v.retain(with_index(|index, item| (index % 2 == 0) || item == 4));

(我更喜欢后者(

Answer 2

我在 Rust 用户论坛上发现了基本相同的问题。他们提出了这个解决方案，这还不错：

let mut index = 0;
v.retain(|item| {
index += 1;
((index - 1) % 2 == 0) || item == 4
});

当时这不是一个有效的解决方案，因为无法保证retain()的迭代顺序，但对我来说幸运的是，该线程中的某个人记录了顺序，所以现在是。

Answer 3

如果你想枚举，过滤(保留(，然后收集结果向量，那么我会说这样做：

v.iter()
.enumerate()
.filter(|&(idx, &val)| val - idx > 0)
.collect()