【66】加一

题目

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

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

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

我的解题

把digits先转换成str，最终转换成int，然后+1，再转换回去成list。

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        numStr=""
        result=[]
        for i in digits:
            numStr=numStr+str(i)
        numInt=int(numStr)+1
        numStr=str(numInt)
        for i in numStr:
            result.append(int(i))
        return result

Accepted

• 111/111 cases passed (24 ms)
• Your runtime beats 97.23 % of python3 submissions
• Your memory usage beats 51.93 % of python3 submissions (14.9 MB)

看了评论区后，才知道大部分情况还需要思考整型溢出问题。但是，今天我知道了，python 的int 是长整型，理论上支持无限大的数，不用考虑它的溢出问题。

题解

直接数组上改变

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        fp=len(digits)-1
        if digits[fp]!=9:
            digits[fp]=digits[fp]+1
        else:
            # 定位最后一个是9的位
            while digits[fp]==9 and fp>=0:
                fp=fp-1
            # 把fp后面的都进一为0
            lens=len(digits)-1-fp
            zeros=[0]*lens
            digits[fp+1:]=zeros
            # 更改进位+1
            if fp<0:
                digits.insert(0,1)
            else:
                digits[fp]=digits[fp]+1
        return digits

Accepted

• 111/111 cases passed (36 ms)
• Your runtime beats 33.74 % of python3 submissions
• Your memory usage beats 9.87 % of python3 submissions (15.1 MB)

总结

直接数组上更改要慢一些，判断条件更多。

学到了：列表切片的赋值，必须要是可迭代的数。若已知长度，和赋值，简单的[0]*10,就可以了。也可以结合numpy，生成一个array，进行赋值。

直接转换类型就行。