题意:
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。 最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。假设除了整数 0 之外,这个整数不会以零开头
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/plus-one
给最末数字加一无非两种结果:
- 结果大于9
此时需要向它的前一位进1,前一位进1也可能导致这样两种结果,直到没有进位数,退出。 - 结果小于等于9,运算直接完毕。
解决办法:
使用carry 保留进位数,
当有进位时,carry=1,这时我们需要向前加1,直到前面的某个数加1不会有进位数。
但是可能会出现加一后数组长度不够,比如999,加1后变为1000,这时在将数组的所有元素遍历后,要判断carry是否还为1,如果还为1,我们就要为数组扩容。
public int[] plusOne(int[] digits) {
//最后一位数字小于9,直接加1
if(digits[digits.length-1]<9){
digits[digits.length-1]+=1;
}
//产生进位时
else{
int ca=1;//进位
digits[digits.length-1]= 0;
for (int i = digits.length-2; i >=0; i--) {
if(digits[i]==9){
digits[i]=0;
}else{
//进位消失时,退出循环
digits[i]+=1;
ca=0;
break;
}
}
//最后判断进位是否存在,是否要为数组扩容
if(ca==1){
int[] res=new int[digits.length+1];
res[0]=1;
for (int i = 0; i <digits.length ; i++) {
res[i+1]=digits[i];
}
return res;
}
}
return digits;
}