1. 程式人生 > 實用技巧 >LeetCode 22 加一

LeetCode 22 加一

給定一個由整陣列成的非空陣列所表示的非負整數,在該數的基礎上加一。

最高位數字存放在陣列的首位, 陣列中每個元素只儲存單個數字。

你可以假設除了整數 0 之外,這個整數不會以零開頭。

示例 1:

輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入陣列表示數字 123。

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入陣列表示數字 4321。

這就是一道數學題。根據題意加一,沒錯就是加一這很重要,因為它是隻加一的所以有可能的情況就只有兩種:

除 99 之外的數字加一;
數字 99。
加一得十進一位個位數為 00 加法運算如不出現進位就運算結束了且進位只會是一。

所以只需要判斷有沒有進位並模擬出它的進位方式,如十位數加 11 個位數置為 00,如此迴圈直到判斷沒有再進位就退出迴圈返回結果。

然後還有一些特殊情況就是當出現 9999、999999 之類的數字時,迴圈到最後也需要進位,出現這種情況時需要手動將它進一位。

/**
 * @param {number[]} digits
 * @return {number[]}
 */
var plusOne = function(digits) {
    for(let i = digits.length - 1; i >= 0; i--) {
        digits[i] ++;
        digits[i] = digits[i] % 10;
        if(digits[i] != 0) return digits;
    }
    digits = new Array(digits.length + 1);
    digits[0] = 1;
    for(let i = 1; i < digits.length; i++) {
        digits[i] = 0;
    }
    return digits;
};