给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入:digits = [1,2,3]
输出:[1,2,4]
解释:输入数组表示数字 123。
示例 2:
输入:digits = [4,3,2,1]
输出:[4,3,2,2]
解释:输入数组表示数字 4321。
示例 3:
输入:digits = [0]
输出:[1]
提示:
1 <= digits.length <= 100
0 <= digits[i] <= 9
思路:两种情况,①容器全部元素都为9,此时容器需要扩展一位,首位为1,其余全部为0;②从容器尾向头遍历,找到第一个不是9的元素,让其加一,是9则置为0。
代码实现:
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int size = digits.size() - 1;
while(size >= 0) //找到第一个不为9的元素
{
if(digits[size] == 9) //是9置0
digits[size--] = 0;
else //第一个不为9,让其加一直接返回结果
{
++digits[size];
return digits;
}
}
digits.insert(digits.begin(), 1); //容器所有元素都为9,扩展一位,向容器首位插入一个1,其余位在上面循环中已经置为0
return digits;
}
};