二、字符串（40）66. 加一

66. 加一

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 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

我的题解：

 

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        i = len(digits)- 1#从个位开始
        num = 0#进位
        while i >= 0:#从右向左
            if i == len(digits) - 1:#个位加1
                digits[i] += 1
            else:#其他位加num这个进位
                digits[i] += num
                num = 0#加完后复位为0
            if digits[i] == 10:#进位只会发生在9-》10
                digits[i] = 0
                num = 1
            i -= 1#左移
            if i == -1 and num != 0:#多了一位，创建新列表，把该位已经其他位一次添加进去
                list1 = list()
                list1.append(num)
                list1.extend(digits)
                digits = list1
        return digits

官方题解：

 

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        n = len(digits)
        for i in range(n - 1, -1, -1):
            if digits[i] != 9:
                digits[i] += 1
                for j in range(i + 1, n):
                    digits[j] = 0
                return digits

        # digits 中所有的元素均为 9
        return [1] + [0] * n