LeetCode——66. 加一

题目

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

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

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

示例 1:

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

示例 2:

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

解题思路 

1、从数组的最后一位往前遍历（不包括第一位），如果后一位元素大于等于10则前一位元素加1，该元素减10（已经进位）。

2、单独判断第一位是否大于等于10，如果大于等于10给一个信号（isOver）。

3、 如果接受到信息（isOver）表示第一个元素需要进位，这时创建长度比原数组大一的数组。该数组第一位为1，剩下的元素从原数组复制。

代码实现

class Solution {
    public int[] plusOne(int[] digits) {
        int N = digits.length;
        Boolean isOver  = false;

        digits[N-1] = digits[N-1] + 1;
        for (int i = N - 1; i > 0; i--) {
            if (digits[i] >= 10) {
                digits[i-1] = digits[i-1] + 1;
                digits[i] = digits[i] - 10;
            }
        }
        if (digits[0] >= 10) {
            isOver = true;
        }
            
        if (isOver) {
            int[] news = new int[N+1];
            news[0] = 1;
            digits[0] = digits[0] - 10;
            for (int i = 1; i < news.length; i++) {
                news[i] = digits[i-1];
            }
            return news;
        }
        return digits;
    
    }
}