Rust循环中可变借用后无法不可变借用的解决方法咨询

问题描述 #

我正在通过Advent of Code学习Rust，遇到一个借用冲突的问题：stack1因pop操作被可变借用，但我想在print_stack中对stacks进行不可变借用，从而在每次修改stacks的循环迭代中打印其状态。尝试克隆对象但未成功，求可行解决方案。

原代码示例：

fn print_stack(stacks: &Vec<Vec<usize>>) -> () {
    // some logic to print the stacks.
    // for now use just println!
    println!("{:?}", stacks);
}

fn main() {
    let mut stacks: Vec<Vec<usize>> = vec![vec![], vec![1,2,3,4]];
    let stack1 = stacks.get_mut(1).unwrap();
    
    while let Some(item) = stack1.pop() {
      // something with item
      // i would like to print every step.
      print_stack(&stacks);
    }
}

核心原因 #

Rust的借用规则禁止同一时间对同一数据存在可变引用和不可变引用。原代码中stack1作为stacks的可变引用，生命周期覆盖了整个while循环，导致循环内无法再对stacks进行不可变借用调用print_stack。

解决方案 #

方案1：缩小可变借用的作用域 #

将stack1的可变借用移至循环内部，每次迭代仅在执行pop时创建可变引用，操作完成后立即释放，这样循环内即可正常对stacks进行不可变借用。

修改后代码：

fn print_stack(stacks: &Vec<Vec<usize>>) {
    println!("{:?}", stacks);
}

fn main() {
    let mut stacks: Vec<Vec<usize>> = vec![vec![], vec![1,2,3,4]];
    
    // 每次循环重新获取可变借用，用完即释放
    while let Some(item) = stacks.get_mut(1).unwrap().pop() {
        // 处理item的逻辑
        print_stack(&stacks);
    }
}

方案2：直接使用索引访问（简化版） #

如果能确保索引1对应的栈始终存在，可直接用stacks[1]替代get_mut，同样将可变访问限制在每次pop操作中：

修改后代码：

fn print_stack(stacks: &Vec<Vec<usize>>) {
    println!("{:?}", stacks);
}

fn main() {
    let mut stacks: Vec<Vec<usize>> = vec![vec![], vec![1,2,3,4]];
    
    while let Some(item) = stacks[1].pop() {
        // 处理item的逻辑
        print_stack(&stacks);
    }
}

方案3：临时克隆（适合小数据场景） #

如果必须在循环外部持有stack1的可变引用，可在打印前克隆整个stacks对象，通过克隆副本避开借用冲突（注意：此方法会带来额外性能开销，仅推荐数据量较小时使用）。

修改后代码：

fn print_stack(stacks: &Vec<Vec<usize>>) {
    println!("{:?}", stacks);
}

fn main() {
    let mut stacks: Vec<Vec<usize>> = vec![vec![], vec![1,2,3,4]];
    let stack1 = stacks.get_mut(1).unwrap();
    
    while let Some(item) = stack1.pop() {
        // 处理item的逻辑
        // 克隆当前stacks状态用于打印
        let stacks_clone = stacks.clone();
        print_stack(&stacks_clone);
    }
}

内容的提问来源于stack exchange，提问作者petto