加一

题目链接

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

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

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

示例 1：

输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。

思路

解法一：

简单的模拟题，从低到高先加1，然后判断是否进位，若没有进位，则直接return，否则，继续下一位加1的运算。 但需要考虑到特殊情况，若循环结束时还没有return，则说明最高位也有进位，在开头添加1

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        n = len(digits)
        for i in range(n - 1, -1, -1): # 逆序遍历
            digits[i] += 1 # 
            if digits[i] <= 9: return digits # 若满足条件，则直接返回
            digits[i] = 0
        digits.insert(0, 1) # 特殊情况
        return digits

解法2：

简单的模拟题，从低到高判断数字是不是9，若不是9，则加1，若是9，则找出从末尾开始的第一个不为9的数，将其加1，然后将末尾的9都置为0，进行返回。 但也需要考虑到特殊情况，若全为9，将输出开头为1，其他元素为0的数组

class solution:
    def plusOne2(self, digits: List[int]) -> List[int]:
        n = len(digits)
        for i in range(n - 1, -1, -1):
            if digits[i] != 9:
                digits[i] += 1
                for j in range(i + 1, n):
                    digits[j] = 0
                return digits
        return [1] + [0] * n