加一--算法练习

leetcode66.加一

题目：

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

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

你可以假设除了整数 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. 末位无进位，则末位加一即可，因为末位无进位，前面也不可能产生进位，比如 45 => 46

2. 末位有进位，在中间位置进位停止，则需要找到进位的典型标志，即为当前位 %10后为 0，则前一位加 1，直到不为 0 为止，比如 499 => 500

3. 末位有进位，并且一直进位到最前方导致结果多出一位，对于这种情况，需要在第 2 种情况遍历结束的基础上，进行单独处理，比如 999 => 1000

分析过后就很明了了，直接上代码

var plusOne = function(digits) {
    const len = digits.length;
    //for循环内这里是第一种情况和第二种情况
    for(let i = len - 1; i >= 0; i--) {
        digits[i]++;
        digits[i] %= 10;
        //这里判断如何某一位不为零，说明不是第三种，那直接return结束循环
        if(digits[i]!=0)
            return digits;
    }
    //这里如果前面没有return那么说明这里是第三种方法，已经进位到最高位了，这时候我们多加一位，全部归化为零，在下方第一位再设为1
    digits = [...Array(len + 1)].map(_=>0);;
    digits[0] = 1;
    return digits;
};