力扣:加一

需求

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

分析

给数组+1, 我们用数学知识对其分析: 我们可以倒序遍历数组, 然后判断数组中的每一位值是否等于9, 如果等于 9 则将其值改为 0 , 继续. 如果不等于 9 那么 加1返回即可.

public int[] plusOne(int[] digits) {
    int length = digits.length;
    int index = length - 1;
    while( index >= 0 ){
        if( digits[index] == 9 ){
            digits[index--] = 0;
        }else{
        	//	不等于9
            digits[index] += 1;
            return digits;
        }
    }
    return digits;
}

执行出错了…

我们发现 参数 [9] 是通过不了的…

其实不光是 [9], 像 [ 9, 9], [9,9,9] 这样的数组内的元素全部是 9 的都不会通过的. 原因是 我们处理的过程中兵没有走分支判断的 下部分, 如下图:

解决办法也很简单, 我们发现如果数组中的所有值都是 9 的话, 都执行了分支语句的上个逻辑, 下个逻辑都没有执行, 这样就造成了一个结果: 数组中的值都变为 0 了, 这显然是不对的.

改进代码如下:

public int[] plusOne(int[] digits) {
    int length = digits.length;
    int index = length - 1;
    while( index >= 0 ){
        if( digits[index] == 9 ){
            digits[index--] = 0;
        }else{
        	//	不等于9
            digits[index] += 1;
            return digits;
        }
    }
    int[] arr = new int[length + 1];
    arr[0] = 1; 
    return arr;
}

解释代码:

public int[] plusOne(int[] digits) {
    int length = digits.length; // 获取数组的长度
    int index = length - 1; // 从数组的最后一个元素开始

    // 从数组的最后一个元素向前遍历
    while (index >= 0) {
        if (digits[index] == 9) { // 如果当前位是9
            digits[index--] = 0; // 当前位置设为0，并移动到前一位
        } else {
            // 如果当前位不是9
            digits[index] += 1; // 将当前位加1
            return digits; // 返回修改后的数组
        }
    }

    // 如果循环完成，说明所有的位都是9，例如 [9, 9, 9]
    int[] arr = new int[length + 1]; // 创建一个新数组，其长度比原数组多1
    arr[0] = 1; // 将新数组的第一位设置为1，其余位默认是0
    return arr; // 返回新数组
}

执行代码:

结尾

以上 是我对这道算法的一些遐想和延伸, 可能不是最优解, 但是算法的优化嘛 本身就是一个思索的过程, 能在这个思索和迭代的过程中有所收获和乐趣就是在成长了, 欢迎大家一起来交流更多的解答…