Leetcode: Plus One

class Solution {
public:
    vector<int> plusOne(vector<int> &digits)
    {
        vector<int>::size_type size = digits.size();
        //当digits的每一位都是9的时候，因为返回的结果位数要增加1，所以做一个特殊处理。
        bool flag = true;
        for (size_t i = 0; i < size; i++)
        {
            if (9 != digits[i])
            {
                flag = false;
            }
        }
        if (flag)
        {
            digits.resize(size + 1);
            digits[0] = 1;
            for (size_t i = 1; i <= size; i++)
            {
                digits[i] = 0;
            }
            return digits;
        }
        //当digits的每一位并不都是9的时候
        //如果最后一位不是9，直接加一返回
        if (digits[size - 1] != 9)
        {
            digits[size - 1]++;
            return digits;
        }
        //如果最后一位是9，则变成0，前面一位加一，如果加一以后变成10，则循环进行这样的处理
        else
        {
            digits[size - 1] = 0;
            digits[size - 2]++;
            for (int i = size - 2; i >= 0; i--)
            {
                if (10 == digits[i])
                {
                    digits[i] = 0;
                    digits[i - 1]++;
                }
            }
        }
        return digits;
    }
};

class Solution 
{
public:
    vector<int> plusOne(vector<int> &digits) 
    {
        vector<int> ret(digits);
        reverse(ret.begin(), ret.end());

        int flag = 1;
        for(int i = 0; i < ret.size(); i++)
        {
            ret[i] += flag;
            //这里的flag的结果要么是0，要么是1
            flag = ret[i] / 10;
            ret[i] %= 10; 
        }

        if (flag == 1)
            ret.push_back(1);

        reverse(ret.begin(), ret.end());

        return ret;
    }
};