66. 加一

题目

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 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)