给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321
int* plusOne(int* digits, int digitsSize, int* returnSize) {
int *r = (int *)malloc(sizeof(int)*(digitsSize + 1)); //加一是为了防止进位溢出
*returnSize = digitsSize; //假设没有最高位进位,则返回结果的长度与原数组相同
r[0] = 0; //多分配了一位内存,先占位
int count = digitsSize; //计数
int temp; //进位变量
if (digitsSize == 1) //如果原数组只有一个值,单独处理
{
if (digits[0] == 9) //一位还会产生进位,直接赋值,并将*returnSize + 1
{
r[0] = 1;
r[1] = 0;
*returnSize += 1;
}
else //否则直接加一,程序结束
r[0] = digits[0] + 1;
}
else //有多位,模仿竖式计算
{
for (int i = digitsSize - 1; i >= 0; i--) //从数组下标的高位开始处理,实际是处理数字的低位
{
if (digits[i] + 1 == 10 && i == digitsSize - 1) //先看数字最低位是否产生进位,产生进位,则结果最低位为0
{
r[count - 1] = 0;
temp = 1; //temp记1代表下一位增 1
}
else if (digits[i] + 1 != 10 && i == digitsSize - 1) //最低位没进位,直接加 1
{
r[count - 1] = digits[i] + 1;
temp = 0;
}
else if (digits[i] + temp == 10) //看除最低位是否产生进位
{
r[count - 1] = 0;
temp = 1;
}
else
{
r[count - 1] = digits[i] + temp;
temp = 0;
}
count--; //往高位移动
}
if (digits[0] == 9 && (digits[1] + temp) == 10) //在这种情况下,会在最高位产生一个进位
{
r[0] = 1;
*returnSize += 1;
}
if (r[0] == 0) //没有产生额外进位,处理掉多占的一位,并缩短结果数组的长度
{
*returnSize -= 1;
for (int i = 0; i<*returnSize - 1; i++)
r[i] = r[i + 1];
}
}
return r;
}```