leetcode_66. 加一

leetcode_66. 加一

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

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

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

示例 1:

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

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

方法一：

• 扩容只会有一种情况：全是9，所以如果有扩容的情况，则只需重新生成一个长度为len+1的数组，且该数组第一个为1，其余全部为0.

    // 如果最后一位小于9，直接最后一位+1，返回
    // 如果最后一位等于9，则相加进位，继续向前判断，依次进位
    // 直到没有进位，返回digits，中间还要改变digits, 
    // 如果走到头了，还是有进位，则需要扩容，但扩容只会出现在999..这样的情况
  

class Solution {
    public int[] plusOne(int[] digits) {
        if(digits == null || digits.length == 0) return digits;

        int len = digits.length;
        if(digits[len - 1] < 9) {
            digits[len - 1] = digits[len - 1] + 1;
            return digits;
        }

        int index = len - 1;
        int carray = 1;
            
        while(index >= 0 && carray != 0) {
            int digit = digits[index];
            int sum = digit + carray;

            if(sum == 10) {
                carray = 1;
                digits[index] = 0;
            }else {
                carray = 0;
                digits[index] = sum;
            }

            index--;
        }

        if(index == -1 && carray == 1) {
            digits = new int[len+1];
            digits[0] = 1;
        }

        return digits; 
    }
}

方法二：

    public int[] plusOne(int[] digits) {
        int len = digits.length;
        for(int i = len - 1; i >= 0; i--) {
            digits[i]++;
            digits[i] %= 10;
            if(digits[i] != 0) {
                return digits;
            }
        }
        digits = new int[len + 1];
        digits[0] = 1;
        return digits;
    }