如何从Rust中的多维数组中交换值？

Question

使用定义为[[f64; 4]; 4]的矩阵（多维固定大小数组），是否可以交换两个值？

std::mem::swap(&mut matrix[i][k], &mut matrix[k][l]); 

给出了错误：

error[E0499]: cannot borrow `matrix[..][..]` as mutable more than once at a time 
    --> math_matrix.rs:100 
    | 
100 | std::mem::swap(&mut matrix[i][j], &mut matrix[k][l]); 
    |      ------------  ^^^^^^^^^^^^^^^- first borrow ends here 
    |      |     | 
    |      |     second mutable borrow occurs here 
    |      first mutable borrow occurs here 

---

我能想出如何做到这一点的唯一方法是使用一个临时的值，例如：

macro_rules! swap_value { 
    ($a_ref:expr, $b_ref:expr) => { 
     { 
      let t = *$a_ref; 
      *$a_ref = *$b_ref; 
      *$b_ref = t; 
     } 
    } 
} 

然后使用：

swap_value!(&mut matrix[i][k], &mut matrix[maxj][k]); 

有没有更好的选择？

Answer 1

您需要使用split_at_mut拆分外层。这就造成了两个不相交的可变引用，然后可以单独交换：

use std::mem; 

fn main() { 
    let mut matrix = [[42.0f64; 4]; 4]; 

    // instead of 
    // mem::swap(&mut matrix[0][1], &mut b[2][3]); 

    let (x, y) = matrix.split_at_mut(2); 
    mem::swap(&mut x[0][1], &mut y[0][3]); 
    //        ^-- Note that this value is now `0`! 
} 

在最一般的情况下，你可能需要添加一些代码，找出哪里拆分和顺序。