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;    }};