题目
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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)+1numStr=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)-1if digits[fp]!=9:digits[fp]=digits[fp]+1else:# 定位最后一个是9的位while digits[fp]==9 and fp>=0:fp=fp-1# 把fp后面的都进一为0lens=len(digits)-1-fpzeros=[0]*lensdigits[fp+1:]=zeros# 更改进位+1if fp<0:digits.insert(0,1)else:digits[fp]=digits[fp]+1return 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,进行赋值。
直接转换类型就行。