在对同一向量进行迭代的同时有效地改变向量

Question

我有一个结构向量，我将向量中的每个元素与其他所有元素进行比较，并且在某些情况下会改变当前元素。

我的问题是你不能同时发生可变和不可变借用，但我不确定如何在不克隆当前元素或整个向量的情况下重新构建我的问题以解决这个问题，这似乎是一种浪费，因为我只对当前元素进行了变异，并且不需要将其与自身进行比较（我跳过了这种情况）。

我确信在 Rust 中有一种惯用的方式来做到这一点。

struct MyStruct {
    a: i32,
}

fn main() {
    let mut v = vec![MyStruct { a: 1 }, MyStruct { a: 2 }, MyStruct { a: 3 }];

    for elem in v.iter_mut() {
        for other_elem in v.iter() {
            if other_elem.a > elem.a {
                elem.a += 1;
            }
        }
    }
}

Answer 1

如果可以修改v的type类型，并且v的元素为Copy，则可以将MyStruct包裹在Cell中。

#[derive(Copy, Clone)] 
struct MyStruct {
    a: i32,
}

fn main() {
    use std::cell::Cell;
    let v = vec![
        Cell::new(MyStruct { a: 1 }),
        Cell::new(MyStruct { a: 2 }),
        Cell::new(MyStruct { a: 3 }),
    ];

    for elem in v.iter() {
        for other_elem in v.iter() {
            let mut e = elem.get();
            if other_elem.get().a > e.a {
                e.a += 1;
                elem.set(e);
            }
        }
    }
}

如果您将 &mut 传递给切片（或可以转换为切片的 &mut），请使用 Cell::from_mut 和 Cell::as_slice_of_cells 并使用与上述相同的技巧（假设切片的元素是 @ 987654327@).

Answer 2

最简单的方法是只使用索引，它不涉及任何长期借用：

for i in 0..v.len() {
    for j in 0..v.len() {
        if i == j { continue; }
        if v[j].a > v[i].a {
            v[i].a += 1;
        }
    }
}

如果你真的，真的想要使用迭代器，你可以通过将 Vec 分成不相交的切片来实现：

fn process(elem: &mut MyStruct, other: &MyStruct) {
    if other.a > elem.a {
        elem.a += 1;
    }
}

for i in 0..v.len() {
    let (left, mid_right) = v.split_at_mut(i);
    let (mid, right) = mid_right.split_at_mut(1);
    let elem = &mut mid[0];

    for other in left {
        process(elem, other);
    }
    for other in right {
        process(elem, other);
    }
}