LeetCode 66.Plus One

1.题目

2.题意

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

3.我的解法

/**
 * 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 carry=1;
    * returnSize=digitsSize;
    for(int i=digitsSize-1;i>=0;i--){
    	digits[i]=digits[i]+carry;
    	if(digits[i]>=10){
    		digits[i]-=10;
    		carry=1;
    	}
    	else{
    		return digits;
    	}
    }
    if(carry){
    	digits=(int*)realloc(digits,(digitsSize+1)*sizeof(int));
    	memmove(digits+1,digits,digitsSize*sizeof(int));
    	digits[0]=carry;
    	* returnSize+=1;
    }
    return digits;
}

整个程序分为两个部分，首先是进位的问题，用carry来表示是否产生新的高位，如果有进新的高位，则需要转入下个部分。

第二部分就是在原来数组空间的基础之上，再分配一个int容量，然后将新的数组整体往后移动一位int，在digits[0]处加入carry值，同时使返回数组大小加一。优于100%

4.优秀解法

int* plusOne(int* digits, int digitsSize, int* returnSize) {
    if (digits == NULL)
    {
        return NULL;
    }
    int n = digitsSize-1;
    while(n >= 0)
    {
        if(digits[n] < 9)
        {
            digits[n]++;
            *returnSize = digitsSize;
            return digits;
        }else
        {
            digits[n] = 0;
            n--;
        }
    }
    
    int* newdigit = (int*)malloc((digitsSize+1) * sizeof(int));
    newdigit[0] = 1;
    for(int i = 1; i < (digitsSize+1); i++)
    {
        newdigit[i] = digits[i-1];
    }
    *returnSize = digitsSize+1;
    return newdigit;
}

与上面解法非常类似，只不过最后采用的是逐位复制的方法，而不是在内存操作，可能在这里花费了一些时间，不如上面的解法