Leetcode 66 Plus One

Leetcode 66 Plus One

#include <vector>
using namespace std;
class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int length = digits.size();
        if (digits[length - 1] < 9){
            digits[length - 1] += 1;
            return digits;
        }
        //碰到第一个不是9的就加1
        for (int i = length - 1; i >= 0; i--){
            if (digits[i] == 9)
                digits[i] = 0;
            else{
                digits[i] += 1;
                return digits;
            }
        }
        //如果前面的情况都没有return，即全为9，会变成全0，需要在最前面插入一个1
            digits.insert(digits.begin(), 1);
            return digits;

    }
};