66. 加一

Posted 2021-01-10 rencoo

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

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

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

// 示例 1:

// 输入: [1,2,3]

// 输出: [1,2,4]

// 解释: 输入数组表示数字 123。

// 示例 2:

// 输入: [4,3,2,1]

// 输出: [4,3,2,2]

// 解释: 输入数组表示数字 4321。

 1/**
 2 * @param {number[]} digits
 3 * @return {number[]}
 4 */
 5
 6var plusOne = function(digits) {
 7    var lastItem = digits[digits.length - 1]
 8    // 最后一位不是 9
 9    if(lastItem !== 9) {
10        digits[digits.length - 1] = lastItem + 1
11    }
12    // 最后一位是 9
13    else {
14        // 全是 9
15        if(new Set(digits).size === 1) {
16            for(var i=0; i<digits.length; i++) {
17                digits[i] = 0
18            }
19            digits.unshift(1)
20        }
21        // 有不是 9
22        else {
23            for(var i=digits.length-1; i>=0; i--) {
24                // 找到不是 9 的
25                if(digits[i] !== 9) {
26                digits[i] = digits[i] + 1
27                break
28            }
29                digits[i] = 0
30            }
31        }
32    }   
33    return digits
34}