leetcode刷题python之加一

思路：加一是需要考虑进位，所以先设置了每一位的进位flag列表，为‘9’则进位，不为9则加一后直接输出

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        n = len(digits)
        flag = [0]*(n - 1)
        for i in range(n-1, -1, -1):
            if i == n - 1:
                if digits[i] == 9:
                    digits[i] = 0
                    if i > 0:
                        print(i)
                        flag[i - 1] = 1
                    else:
                        return [1] + digits
                else:
                    digits[i] += 1
                    return digits
            else:
                print(flag[i])
                print(digits[i])
                if flag[i] == 1:
                    if digits[i] == 9:
                        digits[i] = 0
                        if i > 0:
                            flag[i - 1] = 1
                        else:
                            return [1] + digits
                    else:
                        digits[i] += 1
                        return digits
                else:
                    return digits