原題
給定一個(gè)非負(fù)數(shù),表示一個(gè)數(shù)字?jǐn)?shù)組,在該數(shù)的基礎(chǔ)上+1,返回一個(gè)新的數(shù)組。
該數(shù)字按照大小進(jìn)行排列,最大的數(shù)在列表的最前面。
樣例
給定 [1,2,3] 表示 123, 返回 [1,2,4].
給定 [9,9,9] 表示 999, 返回 [1,0,0,0].
解題思路
- 模擬十進(jìn)制進(jìn)位加法
- 定義一個(gè)carry,初始為1。如carry等于1當(dāng)前數(shù)組為加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