leetcode 加一

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

第一次提交（8ms）：

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
            vector<int> ret = digits;
            int c=0;
            int i = ret.size()-1;
            while(i>=0){
                ret[i]++;
                c = ret[i]/10;
                if(c!=0)
                    ret[i]%=10;
                else return ret;
                i--;
            } 
        if(c!=0)
            ret.insert(ret.begin(),c);
        return ret;  
    }
};

 改进后（0ms）：

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