给定一个非负整数组成的非空数组,在该数的基础上加一,返回一个新的数组。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3] 输出: [1,2,4] 解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1] 输出: [4,3,2,2] 解释: 输入数组表示数字 4321。
本题我刚开始时想着把数组变成一个数,然后再变回数组,碰到 9 这种特殊情况就不行了,
所以参考了大神的代码,发现我真的还差很远很远。
解题思路:分两种情况,若没有进位则数组的尾数直接加 1 返回数组即可;
若有进位,进位进到最前头,则重设另一长度加 1 的数组,数组头存放代表进位的标志,其余就是原数组的减 1 ;
若进位进到前一位,则直接返回 digits。
public static int[] plusOne(int[] digits) {
int len = digits.length -1 ;
int carry = 1;
for(int i = len;i >= 0; --i) {
int number = digits[i] + carry;
digits[i] = number % 10;
carry = number / 10;
if(carry == 0) {
return digits;
}
}
int num[] = new int[len+2];
num[0] = carry;
for(int i = 1;i < len;++i) {
num[i] = digits[i-1];
}
return num;
}