LeetCode 66. 加一（python）

题目链接

题目描述：

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

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

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

解题思路：
在数组最后一位上加1，用flag表示进位以及初始的1，如果当前值加上flag不大于9，说明没有进位，可以返回加1后的数组
如果大于等于10，用flag表示进位
遍历数组后，如果flag不等于0说明最高位产生了进位，需要在数组中添加最高位的进位

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

评论区答案：

        res = []
        n = len(digits)
        for i in range(n-1, -1, -1):
            if digits[i] < 9:
                digits[i] += 1
                return digits
            digits[i] = 0
        
        res = [0 for i in range(n+1)]
        res[0] = 1

        return res