【刷算法】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;
};