LeetCode 66 加一

LeetCode 66 加一

问题描述：给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。你可以假设除了整数 0 之外，这个整数不会以零开头来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/plus-one

输入: [1,2,9]
输出: [1,3,0]
解释: 输入数组表示数字 129。
输入: [9,9,9,9]
输出: [1,0,0,0,0]
解释: 输入数组表示数字 9999。
输入:[1,2,3]
输出:[1,2,4]

头两个示例是两种特殊情况，一个是进位但不加位，一个是进位且加位。分别考虑第一种for循环加1即可，第二种则利用新的长度为输入大小加1的vector 直接生成。

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int digitsSize = digits.size(); //先求出向量的长度
        vector<int> newDigits(digitsSize + 1, 0); //定义一个长度为digits长度加1的新向量,初始化所有的值为0
        for (auto r_iter1 = digits.rbegin(); r_iter1 != digits.rend();)
        {
            *r_iter1 += 1; //利用尾迭代器，当前位加1
            if ((*r_iter1) % 10 == 0) //判断当前位是否为9，为9则进位，否则直接返回结果
            {
                *r_iter1 = 0; //如果是9，则将当前位置0
                if((r_iter1+1) != digits.rend())//判断当前位是不是最高位，不是迭代器加1
                {
                   r_iter1++; 
                }
                else //是，则直接赋值，返回新vector<int>
                {
                    newDigits[0] = 1;
                    return newDigits;
                    break;
                }
            }
            else
            {
                return digits;
                break;
            }
        }
    }
};