leetcode 66. Plus One

题意

在数组中存放一个整数的各个位，然后这个整数加1后返回新的整数数组。

题解

每个位都为9时需要增加一个空间。

代码

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int size = digits.size();
        int carry = 1;
        bool flag = true;
        for(int i = size - 1; i >= 0; i--)
        {
            if(digits[i] != 9)
                flag = false;
            digits[i] += carry;
            carry = digits[i] / 10;
            digits[i] = digits[i] % 10;
        }
        if(flag)
        {
            vector<int> newdigits(size + 1, 0);
            newdigits[0] = 1;
            return newdigits;
        }
        return digits;
    }
};