LeetCode 66 [Plus One]

原题

给定一个非负数，表示一个数字数组，在该数的基础上+1，返回一个新的数组。
该数字按照大小进行排列，最大的数在列表的最前面。

样例
给定 [1,2,3] 表示 123, 返回 [1,2,4].
给定 [9,9,9] 表示 999, 返回 [1,0,0,0].

解题思路

• 模拟十进制进位加法
• 定义一个carry，初始为1。如carry等于1当前数组为加1，判断是否超过9

完整代码

class Solution(object):
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        carry = 1
        for i in range(len(digits) - 1, -1, -1):
            if carry == 1:
                digits[i] += 1
                if digits[i] > 9:
                    carry = 1
                else:
                    carry = 0
                    break
            digits[i] = digits[i] % 10
        if carry == 1:
            digits.insert(0, 1)
        
        return digits