天天看點

力扣-66題 加一(C++)- 數組

題目連結:https://leetcode-cn.com/problems/plus-one/

題目如下:

力扣-66題 加一(C++)- 數組
class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        //最後一位不為9的情況
        if(digits[digits.size()-1]<9){
            digits[digits.size()-1]+=1;
            return digits;
        }

        //最後一位為9的情況
        int i;
        reverse(digits.begin(),digits.end());
        for(i=0;i<digits.size();i++){
            digits[i]+=1;

            if(digits[i]>9) digits[i]%=10;
            else break;

        }

        if(i==digits.size()) digits.push_back(1);
        reverse(digits.begin(),digits.end());

        return digits;
    }
};