问题链接:
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;
}
};
喜欢请点赞、关注、收藏哦~