C语言--加一

加一

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。

示例：

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123

示例：

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321

if (digits[digitsSize - 1] + 1 < 10) {
digits[digitsSize - 1] += 1;
*returnSize = digitsSize;
return digits;
}
int i;
int flag = 1;//flag标志着这个数组是不是全由9组成，1表示是的
for (i = 0; i <digitsSize ; i++) {
if (digits[i] != 9) {
flag = 0;
break;
}
}
if (flag == 1) {//标志着这个数组全由9组成
int* re = (int*)malloc((digitsSize + 1) * sizeof(int));
re[0] = 1;
for (i = 1; i < digitsSize + 1; i++) {
*(re + i) = 0;
}
*returnSize = digitsSize + 1;
return re;
}
else{
digits[digitsSize - 1] += 1;
for (i = digitsSize - 1; i >= 0; i--) {
if (digits[i] == 10) {
digits[i] = 0;
digits[i - 1] += 1;
}
}
*returnSize = digitsSize;
return digits;
}