LeetCode66：加一

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

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

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

示例 1:

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

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

简单做法是直接从后向前循环，不断加1，如果最后最高位仍需要进位，就复制整个数组到新的空间中。

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int n=digits.size();
        int forward=0;
        digits[n-1]+=1;
        if(digits[n-1]==10){
            digits[n-1]=0;
            forward=1;
        }
        for(auto it=digits.rbegin()+1;it!=digits.rend();it++){
            if(forward==0) break;
            *it+=forward;
            if(*it==10) *it=0;
            else forward=0;
        }
        if(forward==1){
            vector<int> ret;
            ret.push_back(1);
            for(auto it=digits.begin();it!=digits.end();it++)
                ret.push_back(*it);
            return ret;
        }
        return digits;
    }
};

这里其实可以直接反向循环的时候压入到新数组，最后再压入头部的1，可以用reverse(ret.begin(),ret.end())反转数组。

附上一个很机智的方法：

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]=0;
            }
            else{
                digits[i]+=1;
                break;
            }
        }
        if(digits[0]==0){
            digits[0]=1;
            digits.push_back(0);
        }
        return digits;
    }
};

作者：hwhu2020
链接：https://leetcode-cn.com/problems/plus-one/solution/c-jian-ji-dai-ma-ni-xu-bian-li-yuan-shu-zu-shuang-/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

可以知道只有在全部位都为9的情况下首部才会进位，这样+1操作后全部位都为0，再把首位改为1，最后压入一个0就可以了。