[C语言][LeetCode][66]Plus One

题目

Plus One
Given a non-negative number represented as an array of digits, plus one to the number.
The digits are stored such that the most significant digit is at the head of the list.

标签

Array、Math

难度

简单

分析

题目意思是给一个整形数组，然后对其进行加1操作，然后返回一个新的数组。这里需要考虑一个数组溢出的问题，即这个整形数组全部为9的时候，加1之后，数组会增加一位。

C代码实现

/**
 * 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) {
    int i = 0;

    if(digits[digitsSize-1] != 9)
    {
        digits[digitsSize-1]++;
    }
    else
    {
        for(i=digitsSize-1; i>=0; i--)
        {
            if(digits[i] == 9)
            {
                digits[i] = 0;
            }
            else
            {
                digits[i]++;
                break;
            }
        }
    }

    if(digits[0] == 0)
        *returnSize = digitsSize+1;  
   else
   {
        *returnSize = digitsSize;
        return digits;
   }
    int *array = (int *)malloc(sizeof(int)*(*returnSize));

    for(i=*returnSize-1; i>=1; i--)
        array[i] = digits[i-1];

    array[0] = 1;   

    return array;
}

上面可以正确提交，下面这种方法却不行，但在PC（ubuntu系统）上可以正确输入输出

/**
 * 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) {
    int i = 0, j=0;
    int *ret;

    for(i=digitsSize-1; i>=0; i--)
        if(digits[i] != 9)
            break;

    if(i == -1)
    {
        ret = (int *)malloc(sizeof(int)*(digitsSize+1));
        memset(ret, 0, sizeof(ret));
        ret[0] = 1;
        *returnSize = digitsSize+1;

        return ret;
    }
    else
    {
        for(j=digitsSize-1; j>i; j--)
           digits[j] = 0;  

        digits[i]++;
        *returnSize = digitsSize;
        return digits;
    }
}

关于上面那个LeetCode提交错误的问题，是因为有一个细节的地方没有注意，memset那里错了，sizeof(ret)求的只是指针的大小，修改如下

   int i = 0, j=0;
    int *ret;

    for(i=digitsSize-1; i>=0; i--)
        if(digits[i] != 9)
            break;

    if(i == -1)
    {
        ret = (int *)malloc(sizeof(int)*(digitsSize+1));
        memset(ret, 0, sizeof(int)*(digitsSize+1));
        ret[0] = 1;
        *returnSize = digitsSize+1;
    }
    else
    {
        for(j=digitsSize-1; j>i; j--)
           digits[j] = 0;  

        digits[i]++;
        *returnSize = digitsSize;
        ret = digits;
    }

    return ret;