给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
这道题可以用两个思路来做,顺便回顾下知识点:
python中数据类型的转换
python中将整形转化成list型,我的思路是将整形转化为str型的,再将str转化成list型的,应该有简洁的方法,这点查到相关办法再做说明。其中需要注意,将list转化为str型的时候不能直接用str()转化,这样转化的结果带[]
例:nums = ['1','2','3']
nums_str = "".join(nums)
这样转化的结果为"123",还有就是需要注意,用这个方法转化需要原来list中的是字符型的,如果原来的list中是int型的,则:
nums = [1,2,3]
nums_str = "".join([str(i) for i in nums])
这样就可以实现需求了。
下面贴上我做这道题用的两种思路解决的代码:
class Solution:
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
digits_str = "".join([str(i) for i in digits])
digits_int = int(digits_str)+1
digits_int = str(digits_int)
digits_list = [int(i) for i in digits_int]
return digits_list
这个是我根据类型转换实现的,代码很简单,但是缺点就是用时很长。
下面讲下我下面这段代码的思路:
其实就是让列表的最后一位加1,然后循环判断一个位置上是否为10,如果为10,就将他的前一位加1,并且这一位变为0,每一位遍历完后,再看第一位,列表的第一个位置上是否为10,然后根据情况再做调整。
class Solution:
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
list_new = digits + [0]
digits[len(digits)-1] += 1
i = len(digits)-1
while i!=0:
if digits[i] == 10:
digits[i-1],digits[i] = digits[i-1]+1,0
i-=1
if digits[0] == 10:
list_new[0],list_new[1],list_new[2:] = 1,0,digits[1:]
return list_new
else:
return digits
这段代码用时较少,其实还可以优化,不用申请额外的空间,但是在时间上的优化的话找到好的方法再做说明。