需求
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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; // 返回新数组
}
执行代码:
结尾
以上 是我对这道算法的一些遐想和延伸, 可能不是最优解, 但是算法的优化嘛 本身就是一个思索的过程, 能在这个思索和迭代的过程中有所收获和乐趣就是在成长了, 欢迎大家一起来交流更多的解答…