LeetCode 66 加一

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/jsszwc/article/details/87890033

题目：

https://leetcode-cn.com/problems/plus-one/

题意：

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

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

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

思路：

直接模拟加法，或者根据只有9+1时才会进位的特性，用一个标记变量记录进位，一旦当前遍历数字不是9，就已经计算完毕了，最后根据标记变量来决定是否增加一位数字

代码：

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        reverse(digits.begin(), digits.end());
        int val = 1;
        for(int i = 0; i < digits.size(); i++) {
            int temp = digits[i] + val;
            digits[i] = temp % 10;
            val = temp / 10;
            if(val == 0) {
                break;
            }
        }

        if(val) {
            digits.emplace_back(val);
        }

        reverse(digits.begin(), digits.end());
        return digits;
    }
};

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        reverse(digits.begin(), digits.end());
        bool flag = false;
        for(int i = 0; i < digits.size(); i++) {
            if(digits[i] != 9) {
                digits[i] += 1;
                flag = false;
                break;
            } else {
                digits[i] = 0;
                flag = true;
            }
        }

        if(flag) {
            digits.emplace_back(1);
        }

        reverse(digits.begin(), digits.end());
        return digits;
    }
};