题目:
给定一个非负整数组成的非空数组,在该数的基础上加一,返回一个新的数组。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
输入: [1,2,3] 输出: [1,2,4] 解释: 输入数组表示数字 123。
输入: [4,3,2,1] 输出: [4,3,2,2] 解释: 输入数组表示数字 4321。
题解:
用变量c记录进位,从列表尾向前遍历,每次当前位+=c,如果移出,当前位=0,c=1;否则结束运算。另外判断是否首位溢出。
class Solution:
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
c = 1
for i in range(len(digits)-1, -1, -1):
digits[i] += c
c = 0
if digits[i]>=10:
c = 1
digits[i] = 0
elif i>0:
return digits
if c:
digits.insert(0, 1)
return digits