天天看点

66. Plus One 把数组看成一个数值进行加1

例如:
Input: [1,2,3]
Output: [1,2,4]
Explanation: The array represents the integer 123.
           

难度:【easy】

把[1,2,3]看着数值123,对它加1,结果就是124.

需要考虑进位情况。

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