加一

题目

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 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

题解

解题思路

将数组从后往前遍历。
如果最后一位不是9的话，直接将最后一位加一，并返回数组；
如果最后一位是9，则先将最后一位变成0，然后继续执行循环;
若能执行到循环结束，则说明每一位都是9，那么只需要在遍历过后的数字开头插入1即可。

/**
 * @param {number[]} digits
 * @return {number[]}
 */
export default function plusOne (digits: number[]): number[] {
  // 空数组直接返回
  if (!digits.length) return []
  // 从后往前遍历
  for (let i = digits.length - 1; i >=0; i--) {
    // 如果最后一位是9，则将它变成0
    if (digits[i] === 9) {
      digits[i] = 0
    } else {
      // 否则直接+1返回
      digits[i]++
      return digits
    }
  }
  // 如果数组中都是9，则会走到这一步
  // 将所有数变成0之后，数组头部插入1即可
  digits.unshift(1)
  return digits
};