题目链接

加一

题目描述

注意点

• 0 <= digits[i] <= 9

解答思路

• 根据数学思想，对最后一位进行加一，任意位置进一会对下一位有影响
• 当digits中所有元素都为9时，整个数字都会加一位，也就是10000…，需要特殊考虑

代码

class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;
        for (int i = n - 1; i >= 0; i--) {
            // 最后一位本身加一，其余位置元素为上一位进一加一
            digits[i]++;
            digits[i] = digits[i] % 10;
            if (digits[i] != 0) {
                return digits;
            }
        }
        // digits中元素都为9，需要增加一位
        digits = new int[n + 1];
        digits[0] = 1;
        return digits;
    }
}

关键点

• digits中所有元素都为9时特殊考虑
• 数组原地操作的过程