Leetcode C语言 “加一”

加一

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。
例:

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

实例2:

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

---

整体思路就是从数组最后一个元素判断是否为9或者0，若为9或0则将该元素变0前一位进一，若不为0，则直接进一。这两个情况都是不用新创建数组的，返回原来的数组。另一种情况就是999这样的数字，需要新创建一个数组(比原数组大一位)，并把新数组里的元素置为0第一位置为1，再返回新数组。
C语言版本实现如下：

#include<stdio.h>
#include<stdlib.h>
//创建一个返回int *的函数指针,因为最后要有一个数组来接受函数的返回值
//int *digits参数接受数组，digitsSize代表数组的大小
int* plusOne(int* digits, int digitsSize, int* returnSize) 
{
    int i;
    //申请了一个比原数组大的数组
    int* ree = (int*)malloc((digitsSize + 1) * sizeof(int));
    for (i = digitsSize - 1; i >= 0; i--)
    {
        if (digits[i] < 9)
        {
            digits[i] = digits[i] + 1; break;
        }
        else
        {
            digits[i] = 0;
        }
    }
    //全为9的情况
    if (digits[0] == 0)
    {
        int j;
        for (j = i; j < digitsSize + 1; j++)
        {
            ree[j] = 0;
        }
        ree[0] = 1;
        *returnSize = digitsSize + 1;
        return ree;
    }
    else
    {
        *returnSize = digitsSize;
        return digits;
    }
    free(ree);
    ree = NULL;

}

int main()
{
    int digits[] = {9,9,9 };
    int returnSize;
    //returnsize是数组的大小，且是全局变量
    int* arr = plusOne(digits, 3, &returnSize);
    for (int i = 0; i < returnSize; i++)
    {
        printf("%d ", arr[i]);
    }
    return 0;
}
``