给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
/*
* 最高位进位:低位数字均为9,需要初始化一个长度为(length+1)的新数组,0位置进1
* 最高位不进位:不需要产生新数组,后续数字由低位计算产生
*/
class Solution {
/*
* 倒序遍历数组进行加1运算,时间复杂度:O(digits.length)
* 存储每次加一的结果,空间复杂度:O(digits.length)
* 针对最高位进位,创建新数组返回,空间复杂度:O(digits.length)
* 执行用时 : 1 ms
* 内存消耗 : 34.8 MB
*/
public static int[] plusOne1(int[] digits) {
int carry=1;
for(int i=digits.length-1;i>=0;i--) {
if(carry==0) {
return digits;//123
}
int tmp;
tmp=digits[i]+1;
carry=tmp/10;
digits[i]=tmp%10;
}
if(carry==0)
return digits;///299
else {
int[] result=new int[digits.length+1];
result[0]=1;
return result;//999
}
}
/*
* 倒序遍历数组直接进行加1运算,时间复杂度:O(digits.length),空间复杂度:O(0)
* 针对最高位进位,创建新数组返回,空间复杂度:O(digits.length)
* 执行用时 : 1 ms
* 内存消耗 : 34.8 MB,
*/
public static int[] plusOne2(int[] digits) {
for (int i=digits.length-1;i>=0;i--) {
if(digits[i]<9) {//123
digits[i]+=1;
return digits;
}
//299
digits[i]=0;
}
int[] result=new int[digits.length+1];
result[0]=1;
return result;//999
}
public static void main(String[] args) {
int nums[]={9,9,9,9};
int result[]= plusOne2(nums);
for(int i=0;i<result.length;i++)
System.out.println(result[i]);
}
}