leetcode-初级-plusone

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

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

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

我的解法：

int* plusOne(int* digits, int digitsSize, int* returnSize) {
    int i,sum=0;
    for(i=0;i<digitsSize;i++)
        sum=sum+digits[i];
    if(digits[digitsSize-1]!=9){
        *returnSize=digitsSize;
        int *newdigits=(int *)malloc((*returnSize)*sizeof(int));
        for(i=0;i<digitsSize-1;i++)
            newdigits[i]=digits[i];
        newdigits[digitsSize-1]=digits[digitsSize-1]+1;
        return newdigits;
    }
    else if(sum==9*digitsSize){
        *returnSize=digitsSize+1;
        int *newdigits=(int *)malloc((*returnSize)*sizeof(int));
        newdigits[0]=1;
        for(i=1;i<=digitsSize;i++)
            newdigits[i]=0;
        return newdigits;
    }
    else
    {
        *returnSize=digitsSize;
        int *newdigits=(int *)malloc((*returnSize)*sizeof(int));
        //long  num=0;
        //int len=digitsSize-1;
        newdigits[digitsSize-1]=digits[digitsSize-1]+1;
        for(i=digitsSize-1;i>0;i--){
            if(newdigits[i]==10){
                newdigits[i]=0;
                newdigits[i-1]=digits[i-1]+1;
            }
            else
                newdigits[i-1]=digits[i-1];
            //printf("%d,%d\n",digits[i],newdigits[i]);
        }
        //printf("运算成功\n");
        return newdigits;
    }

}

答案中时间快的：

​
int* plusOne(int* digits, int digitsSize, int* returnSize) {
	int bit = 1;
	*returnSize = 0;
	int count = 0;
	int carry = 1;
    for(int i = digitsSize - 1;i >=0;i--){
    	(*returnSize)++;
    	if((digits[i]+carry) == 10){
    		count++;
    	}else{
    		carry = 0;
    	}
    }
    if(carry == 1){
    	(*returnSize)++;
    }
    int *ret = (int *)malloc(sizeof(int)*(*returnSize));
    int count1 = (*returnSize) - count;
   	for(int i = (*returnSize)-1;i>=0;i--){
   		if( count > 0){
   			ret[i] = 0;
   			count--;
   		}else if(count == 0 && carry == 1){
   			ret[i] = 1;
   		}else if(count == 0 && i == count1 -1){
   			ret[i] = (digits[i] + 1)%10;
   		}else if(count == 0){
   			ret[i] = digits[i];
   		}
   	}
    return ret;
}

​