lectcode-加一

要求

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

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

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

示例

示例 1:

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

示例 2:

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

代码

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.insert(digits.begin(),1);
    return digits;
}

总结

• 分为三种情况
  • 第一种，末尾数字小于9，直接在最后一位加上1；返回元素组即可。
  • 第二种，末尾存在9，但第一位不是9，如1999，1099，这种情况将为9的哪一位变为0，从后往前第一位不为9的元素加1，然后返回。
  • 第三种，全为9，将所有元素值变为0，再在头部添加1，此时数组长度加长。