习题：加一（C语言/C++）

题目：

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

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

你可以假设除了整数 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]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/plus-one

C语言代码
这是以前还在上C的时候课堂的一道练习题，解法是老师的
灵活运用/=和%=

int* plusOne(int* digits, int digitsSize, int* returnSize)
{
    int tmp = 1;
    for (int i = digitsSize - 1; i >= 0; i--)
    {
        digits[i] += tmp;
        tmp = digits[i] / 10;//判断进位
        digits[i] %= 10;
        if (digits[i] != 0)//代表不进位
        {
            *returnSize = digitsSize;
            return digits;
        }
    }
    int* p = (int*)calloc((digitsSize + 1), sizeof(int));
    p[0] = 1;
    *returnSize = digitsSize + 1;
    return p;
}

下边是自己写的测试

int main()
{
    int arr[] = { 9,9,9,9 };
    int len = sizeof(arr) / sizeof(arr[0]);
    int* returnSize = (int*)calloc(len + 1, sizeof(int));
    int* p = (int*)calloc(len+1, sizeof(int));
    p = plusOne(arr, len , returnSize);
    for (int i = 0; i < *returnSize; i++)
        printf("%d  ", p[i]);
	return 0;
}

C++代码

class Solution
{
public:
    vector<int> plusOne(vector<int>& digits)
    {
        int i = digits.size() - 1;
        digits[i] = (digits[i] + 1) % 10;
        while (i > 0 && digits[i] == 0)
        {
            --i;
            digits[i] = (digits[i] + 1) % 10;
        }
        if ((i == 0 && digits[i] == 0))
        {
            digits.insert(digits.begin(), 1);
        }
        return digits;
    }
};