LeetCode（Python）—— 加一（简单）

加一

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

输入：digits = [1,2,3]
输出：[1,2,4]

输入：digits = [4,3,2,1]
输出：[4,3,2,2]

输入：digits = [0]
输出：[1]

方法一：逆序循环

思路：我们只需要对数组 digits 进行一次逆序遍历，找出第一个不为 99 的元素，将其加一并将后续所有元素置零即可。如果 digits 中所有的元素均为 99，我们需要返回一个新的数组。

话不多说，直接上代码！

# 逆序循环
class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        n = len(digits)
        for i in range(n-1, -1, -1): # 倒序，步长-1
            if digits[i] == 9:
                digits[i] = 0
            else:
                digits[i] += 1
                return digits # 无需进位，直接返回值
        # 所有元素全部为9
        return [1] + [0] * n
class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        n = len(digits)
        for i in range(n-1, -1, -1): # 倒序，步长-1
            if digits[i] != 9:
                digits[i] += 1
                for j in range(i + 1, n):
                    digits[j] = 0
                return digits
        return [1] + [0] * n

总结

奈何老衲没文化，一句C语言走天下！