加一-leetcode-python3

题目：

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

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

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

示例 1:

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

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

 

对于python来说,一行还是可以解决的美观一点的方法

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        return [int(i) for i in str(int(''.join([str(i) for i in digits]))+1)]
       

但是这个题目看来应该不是想考察这种偷懒的办法，应该是处理极大值的时候，数字超出int最大范围的情况，所以就有用数组来存储超出int数字的数据，然后用数组来储存数据无非就是下面几种情况，

第一除了数据头以外有个数据是9，加一就往往前进一位的情况

[1,2,3,4,1,2,3,9,1] or [1,9,9,9,9,9,9,9,9]

第二种数据头是9然后数组需要扩充一位的情况

[9,9,9,9,9,9,9,9,9]

flag = False#设置一个标记值，当为True时候就表明下一个需要进一位
i = len(digits)-1#从数组的最后一个开始
while i >0:
    if digits[i]>=10:#如果当前数字大于10
        flag = True
    if flag:#当前数字大于十
        if i==0:#判断是不是头
            digits.insert(0,1)#是头就在最前面再加一个
        else:
            digits[i-1] += 1#不是头就下一位加一
        digits[i]=0#把当前位数重置为0
        flag = False#重置flag
    i-=1