题目描述:
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。你可以假设除了整数 0 之外,这个整数不会以零开头。
思想:
这回几乎还是用的暴力破解,先把数组转成字符串,再进行加一操作,效率很低
class Solution(object):
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
sum = 0
for i in range(len(digits)):
sum += digits[i]*(10**(len(digits)-i-1))
sum = sum + 1
for i, num in enumerate(str(sum)):
if len(str(sum))>len(digits):
digits.append(0)
digits[i]=int(num)
else:
digits[i]=int(num)
return digits
利用这句来判断是否总位数发生了改变,如果改变了,就给数组增加一位即可
if len(str(sum))>len(digits):
digits.append(0)
digits[i]=int(num)
else:
digits[i]=int(num)
别人效率较高的代码效率提高了些:
class Solution:
def plusOne(self, digits):
#从最后一个元素向前遍历
for i in range(len(digits)-1,-1,-1):
if digits[i]<9:
digits[i]=digits[i]+1
return digits
else:
digits[i]=0
#若以上未返回数组,证明这个数的位数要发生改变,就在最高位加一个1
digits.insert(0,1)
return digits