66.加一Leetcode

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

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

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

示例 1:

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

示例 2:

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

 

class Solution {
    public int[] plusOne(int[] digits) {
        
        if(digits.length == 0 || digits == null){
            return digits;
        }
        
        //当最高位不进位时
        int flag = 1;
        for(int i = digits.length - 1; i >= 0; i--){
            int temp = digits[i] + flag;
            flag = (digits[i] + flag) / 10;//flag为进位
            digits[i] = temp % 10;
            if(flag == 0){
                return digits;
            }
        }
        
        //当最高位进位时，需要初始化一个digits.length+1长的数组
        int[] result = new int[digits.length + 1];
        if(flag == 1){
            result[0] = flag;
        }
        return result;
    }
}

我们可以分为两种情况：

第一种就是当最高位没有进位的时候，只要在原来的数组上进行修改加一后的值即可

第二种情况是当最高位有进位的时候，要重新建立一个新的数组，新的数组的长度比旧数组的长度多1，只有当数组每位都是9的时候才会出现这种情况，这时候，只需要将最高位设置为1即可