题解 | #给数组加一#

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型vector
     */
    vector<int> plusOne(vector<int>& nums) {
        // write code here
        int n = nums.size();
        nums[n-1] += 1;
        int carry = nums[n-1] / 10;
        if(nums[n-1] == 10){
                nums[n-1] = 0;
            }
        for(int i = n-2; i >= 0; i--)
        {
            nums[i] = nums[i] + carry;
            carry = 0;
            if(nums[i] == 10){
                nums[i] = 0;
                carry++;
            }
        }
        if(carry) nums.insert(nums.begin(), 1);
        return nums;
    }
};