&LeetCode0066& 加一

題目

給定一個由整數組成的非空數組所表示的非負整數，在該數的基礎上加一。
最高位數字存放在數組的首位， 數組中每個元素只存儲單個數字。
你可以假設除了整數 0 之外，這個整數不會以零開頭。

示例 1:

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

示例 2:

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

來源：力扣（LeetCode）

思路

首先，判斷最後一位是否爲9，若不是，直接加一返回；若爲9，則給該位賦0，再繼續查前一位，同樣的方法，直到查至第一位；
如果第一位原本爲9，加一後會產生新的一位，那麼查運算完的第一位是否爲0，如果是，則在最前頭加一個1。

C++代碼

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