leetcode 66. 加一

給定一個非負整數組成的非空數組，在該數的基礎上加一，返回一個新的數組。

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

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

解法二：

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

兩種解法的區別在於處理第一個元素需要進位時的操作。

第一種解法是在digits前面插入元素1；

第二種解法是新建一個vec數組，所有值初始化爲0，將第一個元素改爲1。