leetcode66. 加一

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

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

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

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

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

思路:此题分两种情况讨论: 1,当数组最后一个数小于9时,我们直接将它加1就好了。2,当最后一个数等于9时,这时候我们需要将它置为0,然后再循环遍历在它之前的数,看是否还存在9这个数,如果有就将它们置为0,没有的话就跳出循环,再判断此时的nIndex是否小于0,如果小于0,那说明所有的数字都是9,然后加1的结果就是增加一位数字,需要在数组头部插入1,反之,就将当前nIndex所在位置的那个数加1就好了。

class Solution {
public:
    vector plusOne(vector& digits){
        if(digits.empty())
            return digits;
        int nIndex = digits.size()-1;
        if(digits[nIndex] < 9)
        {
            digits[nIndex] += 1;
        }
        else
        {
            digits[nIndex--] = 0;
            while(nIndex >= 0 && digits[nIndex] == 9)
                digits[nIndex--] = 0;
            if(nIndex < 0)
            {
                digits.insert(digits.begin(),1);
            }
            else
            {
                digits[nIndex] += 1;
            }
        }
        return digits;
    }
};

你可能感兴趣的:(Leetcode数组)