题目

给定一个由

整数

组成的

非空

数组所表示的非负整数，在该数的基础上加一。

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

单个

数字。

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

示例 1：

输入：digits = [1,2,3]

输出：[1,2,4]

解释：输入数组表示数字 123。

示例 2：

输入：digits = [4,3,2,1]

输出：[4,3,2,2]

解释：输入数组表示数字 4321。

示例 3：

输入：digits = [0]

输出：[1]

代码

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

解析

分情况讨论

1.末尾没有9的，直接++

2.末尾有9，但是不全是9的，从后向前找到不是9的++，后面的直接赋0

3.全都是9的，直接构造一个n+1长度的数组，第一个为1后面都是0

又是一些小细节

1.虽然说是分情况讨论，但大可不必一上来就if，这样超复杂，最后你会迷失在大括号的层级里（别问我是怎么知道的）

2.解析里的第三点，没有想到可以直接构造新的数组，自己做的麻烦了很多

3.解析中1 2点可以合并，如果末尾没有9直接不走for嘛，是没问题的。

4.总的来说没想到可以这么简单，自己写的又臭又长而且还会卡测试用例

5.c++构造数组： vector < int > ans(n)