LeetCode/066/main.js at master · lazzzis/LeetCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
/**
 * @param {number[]} digits
 * @return {number[]}
 */
var plusOne = function (digits) {
  let ans = []
  let rem = 1
  for (let i = digits.length - 1; i >= 0; i--) {
    let sum = digits[i] + rem
    ans.unshift(sum % 10)
    rem = Math.floor(sum / 10)
    if (rem === 0) {
      ans = digits.slice(0, i).concat(ans)
      break
    }
  }
  if (rem) {
    ans.unshift(rem)
  }
  return ans
}

if (process.env.LZS) { // local test
  const assert = require('assert')
  assert.deepEqual(plusOne([1, 2, 3]), [1, 2, 4])
  assert.deepEqual(plusOne([1, 2, 9]), [1, 3, 0])
  assert.deepEqual(plusOne([9, 2, 9]), [9, 3, 0])
  assert.deepEqual(plusOne([9, 9, 9]), [1, 0, 0, 0])
  assert.deepEqual(plusOne([9]), [1, 0])
}