LeetCode No66. 加一

题目

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

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

你可以假设除了整数 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]

提示：

1 <= digits.length <= 100
0 <= digits[i] <= 9

思路

加法模拟题，挺简单的，只是需要考虑类似于 [9], [9, 9] 这样的数据。

AC代码

点击查看代码

class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;
        int c = 1;
        int[] arr = new int[n+1];
        for(int i=n-1; i>=0; i--) {
            int num = digits[i] + c;
            c = 0;
            if( num>=10 ) {
                c = 1;
                num -= 10;
            }
            arr[i+1] = num;
        }
        if( c == 1 ) {
            arr[0] = 1;
            return arr;
        }
        return Arrays.copyOfRange(arr, 1, n+1);
    }
}