题目
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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,进行赋值。
直接转换类型就行。