原题
给定一个非负数,表示一个数字数组,在该数的基础上+1,返回一个新的数组。
该数字按照大小进行排列,最大的数在列表的最前面。
样例
给定 [1,2,3] 表示 123, 返回 [1,2,4].
给定 [9,9,9] 表示 999, 返回 [1,0,0,0].
解题思路
- 模拟十进制进位加法
- 定义一个carry,初始为1。如carry等于1当前数组为加1,判断是否超过9
完整代码
class Solution(object):
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
carry = 1
for i in range(len(digits) - 1, -1, -1):
if carry == 1:
digits[i] += 1
if digits[i] > 9:
carry = 1
else:
carry = 0
break
digits[i] = digits[i] % 10
if carry == 1:
digits.insert(0, 1)
return digits