思路:加一是需要考虑进位,所以先设置了每一位的进位flag列表,为‘9’则进位,不为9则加一后直接输出
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
n = len(digits)
flag = [0]*(n - 1)
for i in range(n-1, -1, -1):
if i == n - 1:
if digits[i] == 9:
digits[i] = 0
if i > 0:
print(i)
flag[i - 1] = 1
else:
return [1] + digits
else:
digits[i] += 1
return digits
else:
print(flag[i])
print(digits[i])
if flag[i] == 1:
if digits[i] == 9:
digits[i] = 0
if i > 0:
flag[i - 1] = 1
else:
return [1] + digits
else:
digits[i] += 1
return digits
else:
return digits