LeetCode066 Plus One C語言

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.

題意：一個非負的整數用數組的形式保存著。其中高位在a[0]。然后對這個數做加1操作，返回這個數組。

PS:原諒我又么有讀懂題意---！

/**
 * 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;
    int index=0;
    int carry=0;
    int flag=1;
    //感覺是偷了個懶，只有全是9的時候才進1.。。。。。所以
    //只有個位數加1，不是每個都加，所以用flag
    for(i=digitsSize-1;i>=0;i--){
        //不是每個都加1
        if(digits[i]+flag+index>9){
            digits[i]=0;
            index=1;
            if(i==0){
                carry=1;
            }
        }else{
            digits[i]=digits[i]+1;
            break;
        }
        flag=0;
        // printf("%d",digits[i]);
    }    
    //這個值也得寫明白，不然程序不知道？？？
    *returnSize=digitsSize+carry;

    if(carry){
        int *newdigits=(int*)malloc(sizeof(int)*digitsSize+1);
        newdigits[0]=1;
        for(i=1;i<digitsSize+1;i++){
            newdigits[i]=0;
        }
        return newdigits;
    }else{
        return digits;
    }
   
    
}

PS:迷迷糊糊就寫完了。。。。。不執行。。。。。看了一下網上的程序貌似最后還要返回那個returnSize。。。。。。。。。。。。。。這才可以。

其實只有全是9的時候才會產生首位進位。。。。。。。。。。。