【leetcode初级】7-加1

  • 给定一个非负整数组成的非空数组,在该数的基础上加一,返回一个新的数组。最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
    你可以假设除了整数 0 之外,这个整数不会以零开头。

示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

  • 思路:
    相当于数字+1,主要考虑进位的事,如果是9就得在前一位进位,一直到第一位。思路很简单,但是有提交有4个case没通过,发现是因为没考虑到增加位数的问题,即9到10,99到100这种会增加1位的情况。这种特殊情况需要add一个1到最前面。
  • 代码:
class Solution(object):
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        i = len(digits) - 1
        while i >= 0:
            if digits[i] != 9:
                digits[i] += 1
                break
            else:
                digits[i] = 0
                if i == 0:
                    digits.insert(0,1)
                i -= 1
        return digits

你可能感兴趣的:(【leetcode初级】7-加1)