leetcode刷题记录（C语言版）——加一

1. 题目

加一

       给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。你可以假设除了整数 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]

2. 解题思路

• 从数组末尾开始遍历，如果当前元素不是9，直接加1并返回。例如digits = [4,3,2,1]，直接将末尾数加一，得到[4,3,2,2]并返回。
• 如果当前元素是9，将其置为0，继续遍历前一个元素。例如digits = [1,2,3,9,9,9]，那么只需要找到从末尾开始的第一个不为9的元素，将该元素加一，得到[1,2,4,9,9,9]。然后将末尾的9全部置零，得到[1,2,4,0,0,0]并返回。
• 如果所有元素都是9，需要在数组前面添加一个元素，值为1。如[9,9,9,9]，则得[1,0,0,0,0,0]。只需要构造一个长度比digits 多 1的新数组，将首元素置为 1，其余元素置为 0即可。
• 在数组前面添加元素时，需要重新分配内存。
• 最后，输出加1后的数组。

// 函数接受一个整数数组 digits，数组大小 digitsSize，以及一个指向整数的指针 returnSize
int* plusOne(int* digits, int digitsSize, int* returnSize) {
    // 从数组的最后一个元素开始向前遍历
    for(int i=digitsSize-1; i>=0; i--)
    {
        // 如果当前元素不是9
        if(digits[i] != 9)
        {
            // 将当前元素加1
            digits[i]++;
            // 设置返回数组的大小为原数组大小
            *returnSize=digitsSize;
            // 返回原数组，因为不需要扩容
            return digits;
        }
        else {
            // 如果当前元素是9，将其置为0
            digits[i] =0;
        }
    }

    // 如果所有元素都是9，需要创建一个新的数组，大小比原数组多1
    int* digitsnew=(int*) malloc(sizeof(int) * (digitsSize + 1));
    // 新数组的第一个元素设置为1
    digitsnew[0]=1;
    // 新数组的其他元素设置为0
    for(int i=1; i<=digitsSize; ++i)
    {
        digitsnew[i]=0;
    }
    // 设置返回数组的大小为新数组的大小
    *returnSize=digitsSize+1;
    // 返回新数组
    return digitsnew;
}