LeetCode 66. 加一（Javascript）

题目

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

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

示例 2:

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

思路

1. 末尾数字加 1 ，接着从末尾开始向前判断
2. 如果数字等于 10 的话，那么当前数字变为 0 ，并且
   • 前面一个数存在的话，那么前面的数加 1
   • 前面一个数不存在的话，那么直接返回 [1, ...digits]
3. 如果数字不等于 10 的话，直接结束循环，返回 digits

代码

/**
 * @param {number[]} digits
 * @return {number[]}
 */
var plusOne = function(digits) {
    const length = digits.length
    digits[length - 1]++
    for(let i = digits.length - 1; i >= 0; i--) {
        if(digits[i] === 10) {
            digits[i] = 0
            if (i > 0) {
                digits[i - 1] = digits[i - 1] + 1
            } else {
                return [1, ...digits]
            }
        } else {
            break
        }
    }
    return digits
};

复杂度分析

• 空间复杂度：O(N)
• 时间复杂度：O(1)