LeetCode - 加一 【Java | 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) {
        int index = digits.length-1;
        digits[digits.length-1]++;
        
        while(digits[index] == 10){
            digits[index] = 0;
            index--;
            if(index == -1){
                break;
            }
            digits[index]+=1;
        }
        
        if(index == -1){
            int[] ret = new int[digits.length+1];
            ret[0] = 1;
            ret[1] = 0;
            if(digits.length > 1){
                for(int i = 2; i < digits.length;i++){
                    ret[i+1] = digits[i];
                }
            }
            return ret;
        }
        return digits;
    }
}

通过后看见一篇这样的解法，这个申请数组的方式还是第一次见，mark了

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