题目要求
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。 最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。 你可以假设除了整数 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]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/plus-one
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法
循环进位
从数组最后一位开始遍历,如果有进位的话则继续下一位加一,最后需要在循环结束时判断是否还有进位,有的话则需要重新申请内存。
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int len = digits.size();
int end = len - 1;
int carry = 0;
for (int i = end; i >= 0; i--)
{
digits[i] += 1;
if (digits[i] >= 10)
{
carry = 1;
digits[i] = digits[i] % 10;
}
else
{
carry = 0;
break;
}
}
if (carry == 1)
{
vector<int> vtemp;
vtemp.push_back(1);
for (int i = 0; i < len; i++)
{
vtemp.push_back(digits[i]);
}
return vtemp;
}
return digits;
}
};
复杂度分析
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
1
)
O(1)
O(1)