leetcode_66 加一

80 阅读1分钟

要求

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

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

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

核心代码

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

image.png

解题思路:我们先将列表进行翻转,然后对第一个元素进行加1操作,达到10,下一位继续加1,直到最后一个元素,最后一个元素加1为10,则向上进位,在数组最后增加一个1,否则停止。