1. 程式人生 > >(LeetCode每日一刷15)加一

(LeetCode每日一刷15)加一

題目描述:

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

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

你可以假設除了整數 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 len = digits.size();
        vector<int> result(len,0);
        int i;
        bool bPlusOne = 1;
        for (i = len - 1; i >= 0; i--){
            if (bPlusOne){
                if (++digits[i] == 10){
                    result[i] = 0;
                    if (i == 0)
                    {
                        result.insert(result.begin(),1);
                    }
                }
                else{
                    result[i] = digits[i];
                    bPlusOne = 0;
                    continue;
                }
            }
            else{
                result[i] = digits[i];
            }
        }
        return result;
    }
};

 

感想:

1:vectot.size()返回的是一個無符號數,所以在宣告i的型別用decltype(digits.size()) i,這導致了for迴圈成了死迴圈,因為無符號數一直大於0。

2:if後面的花括號不換行我還真不習慣啊。。。