【力扣】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]

提示：

• 1 <= digits.length <= 100
• 0 <= digits[i] <= 9

AC代码

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int len = digits.size();
        vector<int> ans(len + 1);
        int flag;
        if (digits[len - 1] == 9) {
                flag = 1;
                ans[0] = 0;
            }
            else {
                flag = 0;
                ans[0] = digits[len - 1] + 1;
            }
        printf("%d %d",digits[len - 1],ans[0]);
        for (int i = len - 2; i >= 0; i--) {
            if (flag == 1) {
                if (digits[i] == 9) {
                    ans[len - i -1] = 0;
                }
                else {
                flag = 0;
                ans[len - i - 1] = digits[i] + 1;
                }
            }
            else ans[len - i -1] = digits[i];
        }
        if (flag == 1) ans[len] = 1;
        reverse(ans.begin(), ans.end());
        if (ans[0] == 0)
            ans.erase(ans.begin()); 
        return ans;
    }
};