LeetCode刷题java——66.加一(简单)

题目

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

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

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

示例

输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字123

输入：digits = [4,3,2,1]
输出：[4,3,2,2]
解释：输入数组表示数字4321

输入：digits = [0]
输出：[1]

解决方案

从后往前遍历，如果是999这种需要多加一位。

class Solution {
    public int[] plusOne(int[] digits) {
        boolean flag = true;
        int index = digits.length-1;
        while (flag && index >= 0) {
            if(digits[index] == 9) {
                digits[index] = 0;
                flag = true;
            } else {
                digits[index] = digits[index] + 1;
                flag = false;
            }
            index--;
        }

        // 处理到传入数组的第一位，但是例如999这样的例子还需要再往前加一位
        int[] newDigits;
        if(flag) {
            newDigits = new int[digits.length+1];
            newDigits[0] = 1;
            for (int i = 0; i < digits.length; i++) {
                newDigits[i+1] = digits[i];
            }
        } else {
            newDigits = new int[digits.length];
            for (int i = 0; i < digits.length; i++) {
                newDigits[i] = digits[i];
            }
        }
        return newDigits;
    }
}