数组insert_DAY【18】数组

美好的事物 正在悄然发生

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

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        mask = False
        if digits[-1] == 9:
            mask = True
        for i in range(len(digits)-1,-1,-1):
            if mask:
                mask = False
                if digits[i] == 9:
                    mask = True
                    digits[i] = 0
                else:
                    digits[i] += 1
                    break
            else:
                digits[i] += 1
                break
        if mask:
            digits.insert(0,1)
        return digits

力扣​leetcode-cn.com