LeetCode 66. 加一

题目内容

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

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

你可以假设除了整数 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) {
        digits[digits.length-1]++;//先让数组中最后的元素+1

        if (digits[digits.length - 1] == 10) {//判断最后元素+1之后是否等于10，等于就要进位
            for (int i = digits.length - 2; i >= 0; i--) {
                if (digits[i + 1] == 10) {//判断最后元素+1之后是否等于10，等于就要进位
                    digits[i + 1] = 0;
                    digits[i]++;
                }
            }
        }

        if(digits[0]==10){//判断数组第一个元素是否等于10，因为等于10需要进位，所以之前的数组长度不够，必须创建一个新的数组存放
            int[] arr = new int[digits.length+1];
            arr[0]=1;
            arr[1]=0;
            for(int i = 2;i < arr.length;i++){
                arr[i] = digits[i-1];
            }

            return arr;
        }else{//如果数组第一个元素不是10，那就不需要进位，就不需要新创建数组存放了
            int[] arr = new int[digits.length];
            for(int i = 0;i < arr.length;i++){
                arr[i] = digits[i];
            }

            return arr;
        }

    }
}