【刷算法】LeetCode.66-加一

题目描述

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

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

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

示例 1:

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

示例 2:

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

代码实现

/**
 * @param {number[]} digits
 * @return {number[]}
 */
var plusOne = function(digits) {
  let changeLen = false;
  
  for(let i = digits.length-1;i >= 0;i --) {
    let sum = digits[i]+1;
    if(sum > 9) {
      digits[i] = sum%10;
      if(i === 0) {
        changeLen = true;
      }
    }else{
      digits[i] = sum;
      break;
    }
  }
  
  if(changeLen) {
    digits.unshift(1);
  }
  
  return digits;
};