给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。

最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外,这个整数不会以零开头。

示例

输入:digits = [1,2,3]  输出:[1,2,4]

解释:输入数组表示数字 123。

输入:digits = [9,9,9]  输出:[1,0,0,0]

解析

1)数字加一,要从个位数加起,所以倒序遍历

2)个位加1后,如果能整除10,要进1位,以此类推,到不能整除为止

3)如果所有位数加1后都能整除10,最前面要补一位

代码示例

def plusOne(digits):
    for i in range(len(digits)-1, -1, -1):
        digits[i] = (digits[i]+1)%10
        if digits[i]:
            return digits
    digits = [1] + digits
    return digits

digits = [9, 9, 9]
print(plusOne(digits))

执行用时:28 ms, 在所有 Python3 提交中击败了 97.25% 的用户.

本文为 陈华 原创,欢迎转载,但请注明出处:http://ichenhua.cn/read/362