加一：给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。 最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。 你可以假设除了整数 0 之外，这个整数不会以零开头。

LeetCode：加一

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

程序如下：

#include <iostream>
using namespace std;
#include <vector>

vector<int> plusOne(vector<int>& digits) {
    int len = digits.size();
    for (int i = len - 1; i >= 0; i--)
    {
        //如果当前元素不是9的情况
        if (digits[i] < 9)
        {
            //当前元素加一：一种情况是刚开始的加一，另一种是来自于低位的进位
            digits[i]++;
            //由于当前元素加一之后，不会向高位进位，所以直接返回
            return digits;
        }
        //如果当前元素时9，则置当前元素为0，向前进1
        digits[i] = 0;
    }
    //当全部是9的时候，必然位数会增加一位，且首位是1
    vector<int> answer(len + 1);//不赋值，所有默认为0
    answer[0] = 1;
    return answer;
}

int main()
{
    vector < int > v1 = { 1,8 ,9,9 };
    vector < int > v2 = { 9, 9 };
    vector<int> nv1 = plusOne(v1);
    vector<int> nv2 = plusOne(v2);

    for (unsigned i = 0; i < nv1.size(); i++)
        cout << nv1[i] << " ";
    cout << "\n";
    for (unsigned i = 0; i < nv2.size(); i++)
        cout << nv2[i] << " ";

    return 0;
}

结果如下：

1 9 0 0
1 0 0