rust - 如何在不冲突借用的情况下切换向量的两个部分？

Question

我打算在 Vec 中切换元素，但我的解决方案存在所有权问题，我的代码是否完全错误？

给定 [1, 2, 3, 4, 5, 6]，预期输出为 [4, 5, 6, 1, 2, 3]。

fn switch(nums: &mut Vec<i32>, k: i32) {

    let t = k as usize;
    let v1 = &nums[..t];
    nums.drain(t..);
    nums.extend_from_slice(v1);

}

error[E0502]: cannot borrow `*nums` as mutable because it is also borrowed as immutable
 --> src/main.rs:7:5
  |
6 |     let v1 = &nums[..t];
  |               ---- immutable borrow occurs here
7 |     nums.extend_from_slice(v1);
  |     ^^^^^-----------------^^^^
  |     |    |
  |     |    immutable borrow later used by call
  |     mutable borrow occurs here

Answer 1

使用 rotate_left或 rotate_right ，其中 mid 是向量的中间。这将有效地执行预期的切换。

let x = vec![1, 2, 3, 4, 5, 6];
let mid = x.len() / 2;

使用rotate_left:

x.rotate_left(mid);
assert_eq!(&x, &[4, 5, 6, 1, 2, 3]);

使用rotate_right:

x.rotate_right(mid);
assert_eq!(&x, &[4, 5, 6, 1, 2, 3]);

相同的方法可用于普通可变切片。如果要交换的两个分区大小相同但不连续，可以创建两个可变元素的迭代器并交换它们，使用 swap_with_slice ...

let (left, right) = x.split_at_mut(mid);
left.swap_with_slice(right);

... 或者通过一个一个地交换每个元素。

let (left, right) = x.split_at_mut(mid);
for (l, r) in Iterator::zip(left.iter_mut(), right.iter_mut()) {
    std::mem::swap(l, r);
}
assert_eq!(&x, &[4, 5, 6, 1, 2, 3]);

另见:

• How to operate on 2 mutable slices of a Rust array?