LeetCode 66.加一

LeetCode 66.加一

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/plus-one
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

• 思路分析

对数组最后一个元素加一处理，，从后向前遍历，如果有等于10的元素，将此元素赋值为0；对前一位元素加一。如果第一位元素等于10，使用iterator insert( iterator pos, const T& value );在数组第一个元素之前插入一个元素。

• 代码

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

• 结果分析

时间复杂度O(n)

• 问题分析

第一次使用了iterator insert( iterator pos, const T& value );方法。