407 · 加一

设置进位标记f

当前位置loc

class Solution {
public:
    /**
     * @param digits: a number represented as an array of digits
     * @return: the result
     */
    vector<int> plusOne(vector<int> &digits) {
        // write your code here
        int f=1;
        int loc=digits.size()-1;
        while(f==1&&loc>=0){
            digits[loc]+=1;
            f=0;
            if(digits[loc]==10){
                digits[loc]=0;
                f=1;
            }
            loc--;
        }
        if(f==1&&loc<0){
            digits[0]=1;
            for(int i=0;i<digits.size();i++){
                digits[i+1]=0;
            }
            digits.push_back(0);
        }
        return digits;
    }
};