LeetCode-66

加一

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

示例
输入：digits = [4,3,9,9]
输出：[4,4,0,0]
解释：输入数组表示数字 4399。

题解

对于给定的数组，将其各位组合看作一个整数，进行加一操作，可分为以下两种情况。第一种，整数末位为0-8，加1后变为1-9，不影响其他各位；第二种，整数末位为9，加一后变为0，产生进位，此时，整数末尾为9的数位均变为0，第一个不为9的数位加1，若整数各位均为9，则数组长度加1，首位应为1，其余各位均为0。

更多题解

代码

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int n = digits.size();
        for (int i = n - 1; i >= 0; --i) {
        	//从低位到高位找到第一个不为9的元素
            if (digits[i] != 9) {
                ++digits[i];
                //后面若干位均置为0
                for (int j = i + 1; j < n; ++j) {
                    digits[j] = 0;
                }
                return digits;
            }
        }

        //digits中所有的元素均为9
        //初始化n+1个默认值为0的元素
        vector<int> ans(n + 1);
        //首位置为1
        ans[0] = 1;
        return ans;
    }
};