简单

问题描述

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

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

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

示例 1：

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

示例 2：

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

示例 3：

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

提示：

• 1 <= digits.length <= 100
• 0 <= digits[i] <= 9

模拟

数组最后一位加一，标记是否需要进位，需要进位则往前遍历加一，直到不需要进位为止，若最后还需要进位，则在数组前补上一个 $1$。

/**
 * @param {number[]} digits
 * @return {number[]}
 */
var plusOne = function (digits) {
  let carry = 0
  const n = digits.length
  for (let i = n - 1; i > -1; i--) {
    const digit = digits[i] + (carry || i === n - 1 ? 1 : 0)
    carry = digit > 9
    digits[i] = digit % 10
    if (!carry) break
  }
  if (carry) digits.unshift(1)
  return digits
}