力扣(C++): 加一

題目描述

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

最高位數字存放在數組的首位， 數組中每個元素只存儲單個數字。

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

示例 1:

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

示例 2:

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

來源：力扣（LeetCode）
鏈接：加一

思路:
分三種情況討論:

1. 數組最後一位 < 9, 直接最後一個元素加一即可
2. 數組最後一位 = 9, 將最後一位賦爲0, 向前進1, 直至不爲9時停止
3. 數組元素全爲9, 全部賦爲0, 給數組中再加入一個元素0, 首元素賦爲 1

代碼示例:

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int len = digits.size() - 1;
        for (int i = len; i >= 0; --i)
        {
            if (digits[i] == 9)
            {
                digits[i] = 0;
            }
            else
            {
                digits[i]++;
                return digits;
            }
        }
        digits.push_back(0);
        digits[0] = 1;
        return digits;
    }
};