Leetcode每日一题2021.2.26第66题：加一

题目描述

示例

思路和算法

digits 前面加0，从后向前循环，逢9就变0；直到不等于9时，加1，结束循环。最后检查 digits 前面加的 0 是否多余，多余的话就去掉。

代码

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        digits = [0] + digits

        for i in range(len(digits) - 1, -1, -1):
            if digits[i] != 9:
                digits[i] += 1
                break
            else:
                digits[i] = 0

        if digits[0] == 0:
            return digits[1:]
        else:
            return digits