DAY9 Plus One

第 11 屆 iThome 鐵人賽

DAY 9

自我挑戰組

刷題記錄與人生分享系列第 9 篇

11th鐵人賽 leetcode c javascript

keyhouse12

2019-09-10 09:51:48

1746 瀏覽

題目:

https://leetcode.com/problems/plus-one/
給一個陣列把最後一個數字加一，如需要進位把該數字變成零下一進位值加一。

解題思路:

判斷該數字是否大於九需要進位，如需要進位把該數字變為0之後判斷下一個數字是否也需要進位以此循環。如循環到最後一個數字也該進位時，則須建立一個新的陣列來儲存。

C版本:

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;
}

Javascript版本:

var plusOne = function(digits) {
    var result = [];
    var n = digits.length -1 ;
    
    while(n >= 0)
    {
        if(digits[n] < 9)
        {
            digits[n]++;
            return digits;
        }
        else
        {
            digits[n] = 0;
            n--;
        }    
    }
    result[0] = 1;
    for(var i = 0; i<digits.length; i++)
    {
      result[i+1] = digits[i];      
    }
    return result;
};