LeetCode-探索-初级-数组-加一-java

加一

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

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

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

示例 1:

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

示例 2:

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

---

比较简单，通过判断是否进位即可

需要注意的只有999这种情况，比较麻烦

public int[] plusOne(int[] digits) {
        boolean flag = digits[digits.length - 1] + 1 == 10;  //若第一位发生进位
        digits[digits.length - 1] = (digits[digits.length - 1] + 1) % 10;  //更改第一位的值
        if (!flag)  //如果在个位不进位，直接返回
            return digits;
        //由于个位不进位，则之后都不会产生进位

        for (int i = digits.length - 2 ; i >= 0 ; i --)
            if (flag) {  //进位
                if ((digits[i] = (digits[i]  + 1) % 10) == 0)
                    flag = true;  //本次还要进位，继续循环
                else  //本次没有进位，直接返回
                    return digits;
            }
            //end if
        //end for
        //此时还有一种情况，99999
        if (flag) {  //一直到最后一位还有进位
            int[] result = new int[digits.length + 1];
            System.arraycopy(digits, 0, result, 1, digits.length);
            result[0] = 1;
            return result;
        }
        //这种是非999的情况
        return digits;
    }