题目
Given a non-negative integer represented as a non-empty array of digits, plus one to the integer.
You may assume the integer do not contain any leading zero, except the number 0 itself.
The digits are stored such that the most significant digit is at the head of the list.
理解
看到这个题,忍不住就想用递归,虽然递归可能不是最快的方法,但是不用就会觉得蠢啊,代码不够简单啊,于是下面的代码就是结果
代码一
class Solution(object):
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
if digits[0] == 0 and len(digits) == 1:
return [1]
elif digits[-1] == 9:
digits[-1] = 0
return self.addOne(digits, -2)
else:
digits[-1] += 1
return digits
def addOne(self, digits, pos):
if abs(pos) > len(digits):
digits = [1] + digits
elif digits[pos] == 9:
digits[pos] = 0
return self.addOne(digits, pos - 1)
else:
digits[pos] += 1
return digits
就不解释了,很简单的题。问题是提交后发现才打败23.97%的人,用了52ms。我去,别人的代码都是有多快,不行我得改进下,不能用递归。估计是函数调用浪费的时间,我得把addOne去掉。
代码二
class Solution(object):
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
if digits[0] == 0 and len(digits) == 1:
return [1]
pos = -1
while True:
if abs(pos) > len(digits):
return [1] + digits
if digits[pos] == 9:
digits[pos] = 0
pos -= 1
else:
digits[pos] += 1
return digits
我去,这个代码用时86ms,只打败了0.93%的人,虽然代码简短了很多…然后我不太相信的又提交了一遍,变成35ms,打败94.72%的人了[笑哭脸],对此我也是无”法科”说。闲着的我又提交了一次,倒是没变化了,看来第一次提交时不知道发生了什么不科学的事。
代码三
class Solution(object):
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
pos = -1
while abs(pos) <= len(digits):
if digits[pos] == 9:
digits[pos] = 0
pos -= 1
else:
digits[pos] += 1
return digits
return [1] + digits
回头又看了下,其实可以更简单啊,最终应该就是这样了。用时就不放上来了,每次都不一样…
结论
再次证实了,循环和递归可以进行某种程度上的互换,但是千万别随意同时使用,谨以此文纪念今天花费了许多时间解决的leetcode第10题