1. 程式人生 > >【LeetCode】66.加一 (C++)

【LeetCode】66.加一 (C++)

題目描述:

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

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

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

示例 1:

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

示例 2:

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

解題方案:

開始想用一個int,然後因為剛做的大數,覺得並不行,然後就改用在原來陣列基礎上改了。

程式碼:

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