题目描述
示例
思路和算法
digits 前面加0,从后向前循环,逢9就变0;直到不等于9时,加1,结束循环。最后检查 digits 前面加的 0 是否多余,多余的话就去掉。
代码
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
digits = [0] + digits
for i in range(len(digits) - 1, -1, -1):
if digits[i] != 9:
digits[i] += 1
break
else:
digits[i] = 0
if digits[0] == 0:
return digits[1:]
else:
return digits