题目:
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
思路;
先对数组的最后一个数字加1记为sum,sum对10取余的值为更新后的最后一个数字,然后更新carry,如果carry等于0直接返回此数组;如果for循环执行完毕,说明数组的第一个数字必定为1,此时扩容。
代码实现
class Solution {
public int[] plusOne(int[] digits) {
int carry=1; //初始化进制为1
int sum=0;
for(int i=digits.length-1;i>=0;i--){ //从最后一位开始遍历
sum=digits[i]+carry;
digits[i]=sum%10; //更新最后一个数字
carry=sum/10; //更新carry
if(carry==0){ //如果进制为0直接返回数组
return digits;
}
}
//如果for循环执行完毕,说明数组的第一个数字必定为1,此时扩容
int[] num=new int[digits.length+1]; //新建一个数组并扩容
num[0]=1; //对数组的第一个元素赋为1,其他数字默认为0
return num; //返回新的数组
}
}