加一
概述:给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。你可以假设除了整数 0 之外,这个整数不会以零开头。
输入:digits = [1,2,3]
输出:[1,2,4]
输入:digits = [4,3,2,1]
输出:[4,3,2,2]
输入:digits = [0]
输出:[1]
方法一:逆序循环
思路:我们只需要对数组 digits 进行一次逆序遍历,找出第一个不为 99 的元素,将其加一并将后续所有元素置零即可。如果 digits 中所有的元素均为 99,我们需要返回一个新的数组。
话不多说,直接上代码!
# 逆序循环
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
n = len(digits)
for i in range(n-1, -1, -1): # 倒序,步长-1
if digits[i] == 9:
digits[i] = 0
else:
digits[i] += 1
return digits # 无需进位,直接返回值
# 所有元素全部为9
return [1] + [0] * n
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
n = len(digits)
for i in range(n-1, -1, -1): # 倒序,步长-1
if digits[i] != 9:
digits[i] += 1
for j in range(i + 1, n):
digits[j] = 0
return digits
return [1] + [0] * n
总结
奈何老衲没文化,一句C语言走天下!