加一（力扣66）

题目描述

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

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

你可以假设除了整数 0 之外，这个整数不会以零开头。
链接：https://leetcode-cn.com/problems/plus-one

示例 1：

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

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

输入：digits=[0]

输出：[1]

思路解析

考虑最高位会有进位，构造比digits列表多一位的列表digits=[0]+digits。数的最低位个位存在列表最高位。反向遍历遍历列表从个位+1，逢9进1。如果相加位是9，则该位相加结果赋值0，直到相加位不是9结束。相加结束后判断digits[0]如果是0，则返回digits[1:],如果digits[0]是1，说明最高位有进位，返回digits。

代码
 

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