LeetCode-66 加一#

题目#

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。最高位数字存放在数组的首位，数组中每个元素只存储单个数字。你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2: 输入: [4,9,9,9]
输出: [5,0,0,0]
解释: 输入数组表示数字 5000。

解法一#

思路#

将数组中的每一个字符连接成一个字符串；然后字符串转换为整数，再将数字增加1；分割数字，组成另一个数组

代码#

1
class Solution:
2
    def plusOne(self, digits: List[int]) -> List[int]:
3
        num = ""
4
        for i in digits:
5
           num += str(i)
6
        num = str(int(num) + 1)
7

8
        ans = []
9
        for i in num:
10
            ans.append(int(i))
11

12
        return ans

复杂度#

空间：O(n) 时间：O(n)

解法二#

思路#

和第一种思路类似遍历数组，将数组变成数字形式新数字加一分割数字，组成另一个数组

代码#

1
class Solution:
2
    def plusOne(self, digits: List[int]) -> List[int]:
3
        intNum = 0
4
        for i in range(len(digits)):
5
            intNum = intNum*10 + digits[i]
6
        intNum += 1
7

8
        res = []
9
        for i in str(intNum):
10
            res.append(int(i))
11

12
        return res

复杂度#

空间：O(n) 时间：O(n)

解法三#

思路#

换个思路，加一无非两个情况

末位为 9 需要进位
末位不为 9，直接加一

首先设边界条件，若无数字则为一
末位为 9：

截取除最后一位外的数组，末位加一
新数组结尾填零

末位不为 9：末位直接加一

代码#

1
class Solution:
2
    def plusOne(self, digits: List[int]) -> List[int]:
3
        if len(digits) == 0:
4
                digits = [1]
5
        elif digits[-1] == 9:
6
            digits = self.plusOne(digits[:-1])
7
            digits.extend([0])
8
        else:
9
            digits[-1] += 1
10
        return digits

为了好理解些，解释下代码 digits = self.plusOne(digits[:-1]) 示例：

1
digits = [9, 0, 9]
2
# digits = [9, 0, 9]
3
digits = self.plusOne([9, 0]) # this returns [9, 1]
4
# digits = [9, 1]
5
digits.extend([0])
6
# digits = [9, 1, 0]