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。

解答:

簡單題,注意判斷進位。

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