leetcode-数组-简单-加一

题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/plus-one
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

解答

方法一：

这个题目主要就是考虑一个进位的问题，我的想法是这样的，只有某位的数字是9才会发生进位，不是9是不会进位，9进位之后就变成了0。那么这个问题就变成，寻找第一个不是9 的数字。如果找完整个数组都没有找到不是9的，那么直接增加一位。找到第一个不是9的数字之后对这个数字 + 1，把后边的数字全部变成0就可以了。

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        ///找到第一个不为9的数，然后把这个数+1 后边的数据全部变成0
        int i = digits.size() - 1;
        for (; i >= 0;) {
            if (digits[i] == 9) {
                digits[i] = 0;
                i--;
            } else {
                digits[i] += 1;
                break;
            }
        }     
        if (i == -1) {
            vector<int> ret(digits.size() + 1);
            ret[0] = 1;
            return ret;
        } else {
            return digits;
        }
    }
};