leetcode66. 加一

题目描述：

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每一个元素只存储一个数字。

你能够假设除了整数 0 以外，这个整数不会以零开头。

示例：

输入：[4,3,2,1]

输出：[4,3,2,2]

#include<iostream>
#include<vector>
using namespace std;

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

int main(){
    Solution solution;
    vector<int> vec;
    int i;
    do{
        cin>>i;
        vec.push_back(i);
    }while(getchar() != '\n');
    vec = solution.plusOne(vec);
    for(int i=0; i<vec.size(); ++i){
        cout<<vec[i]<<" ";
    }
    cout<<endl;
    return 0;
}