Rust中Vec追加报错问题：原因分析与修复方法

问题重现 #

在使用Rust解决LeetCode加一问题时，编写了如下代码：

fn plus_one(digits: &mut Vec<u32>) -> &Vec<u32> {
    for i in (0.. digits.len()).rev() {
        if digits[i] < 10 {
            digits[i] = digits[i] + 1;
            break;
        }else{
            digits[i] = 0;
        }
    }

    if(digits[0] == 0) {
        let mut new_digits : Vec<u32> = vec![1];
        new_digits.extend(digits).        // This is the error
    }

    digits
}

运行后出现如下错误：

--> src/easy/plus_one.rs:13:20
   |
13 |         new_digits.extend(digits)
   |                    ^^^^^^ the trait `Extend<&mut u32>` is not implemented for `Vec<u32>`
   |
   = help: the following other types implement trait `Extend<A>`:
             <Vec<T, A> as Extend<T>>
             <Vec<T, A> as Extend<&'a T>>

错误原因 #

1. 类型不匹配：digits是&mut Vec<u32>类型，直接传给extend时，迭代器产生的元素是&mut u32，但Vec<u32>仅实现了Extend<u32>和Extend<&u32> trait，不支持Extend<&mut u32>，因此触发报错。
2. 语法错误：new_digits.extend(digits)末尾多了一个多余的点.，属于无效语法。
3. 逻辑错误：即使修复前两个问题，当前代码创建new_digits后并未返回它，函数最终还是返回原digits，无法处理全9的情况（比如输入[9,9]，处理后变成[0,0]，此时应该返回[1,0,0]，但原代码做不到）。

修复方案 #

加一操作可能需要创建新数组，返回Vec<u32>比返回引用更合理（符合LeetCode题目的常规要求），修复后的代码如下：

修复后代码 #

fn plus_one(mut digits: Vec<u32>) -> Vec<u32> {
    for i in (0..digits.len()).rev() {
        if digits[i] < 9 {  // 改为<9更准确，避免冗余判断
            digits[i] += 1;
            return digits;
        }
        digits[i] = 0;
    }

    // 走到这里说明所有位都是9，需要创建新数组
    let mut new_digits = vec![1];
    new_digits.extend(digits);
    new_digits
}

关键修复点 #

• 调整函数签名：参数改为mut digits: Vec<u32>，返回Vec<u32>，避免引用带来的生命周期和所有权问题，同时支持返回新数组。
• 优化判断逻辑：将digits[i] <10改为digits[i] <9，因为只有小于9的位加1后不会进位，可直接返回结果，逻辑更严谨。
• 修正语法问题：移除extend语句末尾的多余点.。
• 补全逻辑闭环：在全9的情况下，直接返回创建的new_digits，确保正确生成进位后的新数组。
• 解决类型问题：此时digits是Vec<u32>类型，迭代器产生的是u32元素，符合Vec<u32>的Extend<u32>实现，不会再触发类型错误。

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