Leetcode: Plus One

Given a number represented as an array of digits, plus one to the number.

比较简单，写的时候有点担心全部为9时的时间复杂度，不过通过了。

class Solution {
public:
    vector<int> plusOne(vector<int> &digits) {
        int size = digits.size();
        if (size == 0) return digits;
        
        if (digits[size-1] != 9) {
            digits[size-1] += 1;
        }
        else {
            int last_not_nine = size - 2;
            while (last_not_nine >= 0 && digits[last_not_nine] == 9) {
                --last_not_nine;
            }
            if (last_not_nine >= 0) {
                ++digits[last_not_nine];
            }
            else {
                digits.resize(size+1);
                digits[0] = 1;
                ++last_not_nine;
                ++size;
            }
            for (++last_not_nine; last_not_nine < size; ++last_not_nine) {
                digits[last_not_nine] = 0;
            }
        }
        
        return digits;
    }
};

===========================第二次=========================

直观的方法。

class Solution {
public:
    vector<int> plusOne(vector<int> &digits) {
        int i = digits.size() - 1;
        for (; i >= 0; --i) {
            if (digits[i] != 9) {
                ++digits[i];
                break;
            }
            else {
                digits[i] = 0;
            }
        }
        
        if (i < 0) {
            digits.insert(digits.begin(), 1);
        }
        
        return digits;
    }
};