加一
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
解题思路
将数组从后向前遍历,若当前元素是第一个元素切该元素大于9,将当前元素置0并在数组头插入元素1(进位操作,若整数为[9,9],进位后为[1,0,0]);若当前元素不是第一个元素且其值大于9,当前元素置0且上一个元素加1(进位操作,若整数为[1,9],进位后为[2,0])。
Python代码
class Solution:
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
k = len(digits) -1
digits[k] += 1
i = k
while i >= 0:
if i==0 and digits[i]>9:
digits[i] = 0
digits.insert(0,1)
if i!=0 and digits[i] > 9:
digits[i] = 0
digits[i-1] += 1
i -= 1
return digits