题目:给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
答案:
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
n = len(digits)
while n > 0:
if digits[n-1] == 9 :
digits[n-1] = 0
n -= 1
else:
digits[n-1] = digits[n-1] + 1
return digits
if n == 0:
return [1] + digits
考虑最后一位是否为9,如果不是9,直接加一 ,返回dig即可,如果是9,将其变为0,再考虑前一位是否为9,以此类推,直到第一位,如果dig[0]也是9,那么返回[1]+dig即可。