《leetCode》：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.

题目大意：将用数组表示的一个数进行加1操作。其中数组的第一位是最高位。

思路

此题比较简单，直接进行加1操作即可。要考虑的是：最高位有进位

/**
 * 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) {
    if(digits==NULL||digitsSize<1){
        return NULL;
    }

    int ci=0;
    int i=digitsSize-1;
    for(;i>=0;i--){
        if(i==digitsSize-1){
            ci=(digits[i]+1)/10;
            digits[i]=(digits[i]+1)%10;

        }
        else{
            if(ci==0){
                break;
            }
            int temp=digits[i]+ci;
            ci=(temp)/10;
            digits[i]=(temp)%10;

        }


    }
    if(ci==0){
        *returnSize=digitsSize;
        return digits;
    }
    //最高位有进位
    if(i<0&&ci>0){
        int *result=(int *)malloc((digitsSize+1)*sizeof(int));
        if(result==NULL){
            exit(EXIT_FAILURE);
        }
        result[0]=ci;
        for(int j=1;j<digitsSize+1;j++){
            result[j]=digits[j-1];
        }
        *returnSize=digitsSize+1;
        return result;
    }
}