给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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]
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* plusOne(int* digits, int digitsSize, int* returnSize) {
int i;
int* new=(int*)malloc(sizeof(int)*(digitsSize+1));
int carry=1;
for(i=digitsSize-1;i>=0;i--)
{
int sum=digits[i]+carry;
if(sum>=10)
{
new[i+1]=sum-10;
carry=1;
}
else{
new[i+1]=sum;
carry=0;
}
}
if(carry==1)
{
new[0]=carry;
*returnSize=digitsSize+1;
}
else{
for(int j=0;j<digitsSize;j++)
{
new[j]=new[j+1];
}
*returnSize=digitsSize;
}
return new;
}