原題
給定一個非負數,表示一個數字數組,在該數的基礎上+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