Leetcode 66 Plus One

题目要求：
Given a non-negative number represented as an array of digits, plus one to the number.

The digits are stored such that the most significant digit is at the head of the list.
----------------------------------------------------------------------------------------------------
给定一个非负整数，对其加1，要应对所给数组中每个数字是否是9以及9出现在不同位置有不同处理方法。
从后往前遍历数组每个数字
是9，所在位置：
（1）在末尾，并且不是首位，就将其改为0，继续遍历前面数字；
（2）在末尾，并且是首位，如999,99，要扩大数组容量，建立新数组，改为1000，或100；
（3）只有一个9，即所给输入就是9，那就输出10；
非9，所在位置：
（1）在末尾，直接末尾为加1，返回数组；
（2）在中间，该为加1，返回数组；

代码如下：

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int carry = 0;
        //判断末尾是否是9
        int len = digits.size();
        if(digits[len-1] != 9)
        {
            digits[len-1]++;
            return digits;
        }
        else    //末尾是9
        {
            digits[len-1] = 0;
            carry = 1;
            if(len == 1)    //如果所给输入就是9，返回1,0
            {
                vector<int> res;
                res.push_back(1);
                res.push_back(0);
                return res;
            }
            for(int i = 0; i < len - 1; i++)
            {
                if(digits[len-2-i] != 9)    //某一位不是9，直接加1返回
                {
                    digits[len-2-i]++;
                    return digits;
                }
                else if(digits[len-2-i] == 9 && len-i-2 == 0)   //某一位是9，并且是首位，应对如99,999等数
                {
                    vector<int> res;
                    for(int j = 0; j < len+1; j++)
                    {
                        if(j == 0)
                            res.push_back(1);
                        else
                            res.push_back(0);
                    }
                    return res;
                }
                else if(digits[len-2-i] == 9 && len-i-2 != 0)   //某一位是9，但不是首位，直接将该位赋值为0
                {
                    digits[len-2-i] = 0;
                    continue;
                }
            }
        }
        return digits;
    }
};

个人看了个比较简练的方法，遍历每一位，new_val = digits[i] + carry，如果该位>9，该位-10，并且进位位置1；该位<9，则进位位置0。不管怎样，都将new_value  push_back到新数组中，最后将新数组逆序，得到结果。

代码如下：

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        vector<int> ret;
        if (digits.size() == 0) { ret.push_back(0); return ret;}
        int add = 1;
        for (int i = digits.size()-1; i >= 0; i--) {
            int s = digits[i] + add;
            if (s > 9) {
                add = 1;
                s -= 10;
            } else {
                add = 0;
            }
            ret.push_back(s);
        }
        if (add > 0) {
            ret.push_back(add);
        }
        reverse(ret.begin(), ret.end());
        return ret;
    }
};