Plus One

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize) {
    for(int i = 0; i < digitsSize; i++)
    {
        if(digits[digitsSize - i - 1] == 9)
            {
                digits[digitsSize - i - 1] = 0;
                if(i == digitsSize)
                   {
                        returnSize = malloc((digitsSize + 1) * sizeof(int));
                        memset(returnSize, 0, sizeof(returnSize));
                        returnSize[0] = 1;
                        return returnSize;
                   }
                   
            }
        else
            {
                digits[digitsSize - i – 1] += 1;
                returnSize = malloc((digitsSize) * sizeof(int));
                memcpy(returnSize, digits, digitsSize*sizeof(int));
                return returnSize;
            }
    }
}