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

思路：

这里需要判断的就是进位，有个特殊点，就是每一位都需要进位如999，那么后面补0，前面第一位变成1即可，将数组变长1位

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        for (int i = digits.size()-1; i >=0; i--)
        {
            //循环如果遇到9就进位，否则就+1
            if(digits[i]==9)
                digits[i]=0;
            else{
                digits[i]++;
                break;
            }
        }
        //由于进位导致第一位变为0，后面补个0，第一位改成1
        if (digits[0]==0)
        {
            digits.push_back(0);
            digits[0]=1;
        }
        return digits;
    }
};