LeetCode.66.加一

题目

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

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

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

示例 1:

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

示例 2:

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

思路：

当数组最后一位不是 9，直接在最后一位加上 1，返回整个数组就 ok；否则从后往前遍历整个数组，首先持有进位为 1，依次判断每一位与之前进位相加结果，把结果插入答案数组，最后如果进位还为 1，还得再插入一个。

代码：

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        if(digits.back()!=9){
            digits.back()++;
            return digits;
        }
        vector<int> ans;
        int jw=1,t,i;
        for(i = digits.size()-1 ; i>=0;i--){
            t = digits[i]  + jw;
            jw = 0;
            if(t > 9){
                jw = t/10;
                t %=10;
            }
            ans.insert(ans.begin(),t);
        }
        if(jw)
            ans.insert(ans.begin(),jw);
        return ans;
    }
};