Solution 1
这个题主要是考察加法逻辑和数据结构操作,没有什么难度,就是额外注意一下最后一次加法(头部位置)如果有进位需要额外增加一个元素。为了不增加额外的空间占用,我使用了一种很拧巴的方法实现了vector的头部插入。
- 时间复杂度: O ( n ) O(n) O(n),其中 n n n位输入列表的长度,线性遍历
- 空间复杂度: O ( 1 ) O(1) O(1),只维护常数个状态变量
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int carry = 1;
for (int i = digits.size() - 1; i >= 0; --i) {
int temp = digits[i] + carry;
digits[i] = temp % 10;
carry = temp / 10;
}
if (carry != 0) {
int temp = carry;
for (int i = 0; i < digits.size(); ++i) {
int temptemp = digits[i];
digits[i] = temp;
temp = temptemp;
}
digits.push_back(temp);
}
return digits;
}
};
Solution 2
Solution 1的Python实现
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
carry = 1
for i in range(len(digits) - 1, -1, -1):
temp = digits[i] + carry
digits[i] = temp % 10
carry = temp // 10
if carry != 0:
temp = carry
for i in range(len(digits)):
temptemp = digits[i]
digits[i] = temp
temp = temptemp
digits.append(temp)
return digits