66. Plus One

Problem:

Given a non-negative integer represented as a non-empty array of digits, plus one to the integer.

You may assume the integer do not contain any leading zero, except the number 0 itself.

The digits are stored such that the most significant digit is at the head of the list.

Examples:

Input: [9] output: [1,0]

Input: [1,0,9] output: [1,1,0]

这题我也想吐槽一下英文题目，讲得并不清楚，只有自己去输入好一些样例去测试才知道它的意思，题意是这个数组里面每一位都是一个(0,9)的数，然后往最后一位+1，所以一旦某一位是9了，加1之后就得进位。我新开一个vector来保存值，然后从给出的数组中倒着来计算，设flag=1为需要进位，flag=0则不需要。当某一位是9，且需要进位时，则flag=1，然后该位等于0；当某一位不是9，且需要进位时，则该位等于该位+1；假如不需要进位，直接就是该位的值。到最后，假如flag还等于1，则说明第一位是9而且需要进位，然后再插入一个1。然后用reverse函数把新开的vector倒一下就是结果了。

Code：

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