LeetCode 66. Plus One

题目

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