给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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,返回一个长度为2的数组{ 1,0 }。数组长度大于1时,从最后一位开始加一,满十进一,从最后一个元素向前遍历,如果最后一个元素加一小于10,将加一之后的数组返回;如果等于10,再依次向前进行判断,直到某一位小于10。值得注意的是,如果所有的元素都为9,到达第一个元素时,加一后得10,数组需要扩容一位,所有元素后移一位,首位置1。如果不嫌麻烦,也可以遍历该数组,如果每个元素都为9,直接赋值计算后的结果。
class Solution {
public int[] plusOne(int[] digits) {
int l = digits.length;
int newl=l+1;
int[] newarr = new int[newl];
int i;
if(l==1){
if(digits[0]+1<10){
digits[0]++;
return digits;
}
else{
newarr[0]=1;
newarr[1]=0;
return newarr;
}
}
else{
for(i=l-1;i>0;i--){
digits[i]++;
if(digits[i]<10){
return digits;
}
else{
digits[i]=0;
}
}
}
if(digits[0]+1<10){
digits[0]++;
return digits;
}
else{
digits[0]=0;
for(i=l-1;i>0;i--)
{
newarr[i]=digits[i-1];
}
newarr[0]=1;
}
return newarr;
}
}