Rust每日一题(2)---数据结构-数组-plus-one

leetcode地址 给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1：

输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。

示例 2：

输入：digits = [4,3,2,1]
输出：[4,3,2,2]
解释：输入数组表示数字 4321。

示例 3：

输入：digits = [0]
输出：[1]

知识点

• 数组索引和insert操作
• 整数溢出(i -=1)

思路

如果先数组求和变为整数再加1会存在整数溢出问题(10 as i32).pow(n)，因此需要考虑直接操作数组。

• 1.末尾数非9，直接加1返回
• 2.末尾数为9，将末位设为0，同时末位索引i-1
• 3.全部计算完了还没返回，说明全部都是9，需要增加一位并设为1。

impl Solution {
    pub fn plus_one(mut digits: Vec<i32>) -> Vec<i32> {
        let mut i = digits.len();
        while i > 0 {
            if digits[i-1] !=9 {
                digits[i-1]  += 1;
                return digits
            }
            digits[i-1] = 0;
            i -= 1;
        }
        digits.insert(0, 1);
        digits
    }
}