leetcode 66. 加一（cpp）

问题链接：
https://leetcode.cn/problems/plus-one/

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

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

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

解题思路：
这道题的目标在实现加法。首先反转向量，注意std命名空间和algorithm库的导入。然后遍历反转的向量，即从低位到高位遍历。如果当前位置不是9则不需要进位，所以直接break跳出循环。如果当前位置是9且为原数最高位（代码中的else if情况）则在向量末尾添加1。这种情况的的一个例子是99+1=100。如果当前位置为9但是不是原数的最高位，就将当前位归零然后继续循环。循环结束后再一次反转向量，输出答案。

解题代码：

#include <algorithm>

class Solution
{
public:
    vector<int> plusOne(vector<int> &digits)
    {
        // basically implement add
        std::reverse(digits.begin(), digits.end());
        for (int unit = 0; unit < digits.size(); ++unit)
        {
            if (digits[unit] != 9)
            {
                ++digits[unit];
                break;
            }
            else if (unit + 1 == digits.size())
            {
                digits.push_back(1); // e.g. 99+1=100
                digits[unit] = 0;
                break;
            }
            else
            {
                digits[unit] = 0;
            }
        }
        std::reverse(digits.begin(), digits.end());
        return digits;
    }
};

喜欢请点赞、关注、收藏哦~