LeetCode - 解题笔记 - 66 -Plus One

Solution 1

这个题主要是考察加法逻辑和数据结构操作，没有什么难度，就是额外注意一下最后一次加法（头部位置）如果有进位需要额外增加一个元素。为了不增加额外的空间占用，我使用了一种很拧巴的方法实现了vector的头部插入。

• 时间复杂度：$O(n)$，其中$n$位输入列表的长度，线性遍历
• 空间复杂度：$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