题目描述:
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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
关键思想:“+1后%10”,这样,小于9的数字可以直接+1,而9又可以在+1后,发现等于10,然后%10又置为了0,这样就省去了很多的判断。
重点在于两种特殊情况,1. 连续的9,2. 数组每个元素均是9。
1. 如何让几个连续的9置为0,并且第一个不为9的数字+1?
我们设置一个循环,循环的结束条件是当前位%10后不等于0。等于0就表示这一位为9,那么还需要循环一次,让下一位+1;如果不等于9,那么就直接可以跳出循环,然后将数组返回了。
2. 数组元素每个元素都是9怎么办?
Java是不允许改变一个定长数组的长度的。前面已经知道了,只要当前位+1%10后不等0,那么就会将当前数组返回,而全是9呢,数组一直循环到第一位,将数组第一位置为0后,退出循环了,这个时候还不能返回数组。距离真正的结果,只差了一位1,那么我们只需要用原数组的长度+1来作为新数组的长度,然后将新数组的首位置为1即可。
图解:
1. 连续的9
2. 每个元素都是9
代码实现:
public static int[] plusOne(int[] digits) {
for (int i = digits.length - 1; i >= 0; i--) {
digits[i]++;
digits[i] = digits[i] % 10;
if (digits[i] != 0) return digits;
}
digits = new int[digits.length + 1];
digits[0] = 1;
return digits;
}