66. 加一

文|Seraph

01 | 问题

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

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

示例 2:

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

02 |解题

初解：

从后往前对加一计算模拟，等于10则进位。需要注意的是，如果第一个数发生进位，需要在动态数组最前面插入一个1。

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

终解：

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int carry = 1;
        int length = digits.size();
        for(int i = length - 1; i >= 0; --i)
        {
            int cur_sum = digits[i] + carry;
            digits[i] = cur_sum % 10;
            carry = cur_sum / 10;
        }
        if(carry > 0)
            digits.insert(digits.begin(), 1);
        return digits;
    }
};

03 | 积累知识点

1. vector为动态数组，可以使用insert操作。

官网链接 Leetcode No.66