leetcode 66. 加一

一、题目

二、解答

1.思路

分三种情况

1. 个位数加1小于10不需要进1，则个位数加一后返回数组即可

2. 需要进1。又分为两种情况：1不需要增加最高位，如899 变为900；2需要增加一位，如999变为1000

   1. 第一种情况，不需要增加最高位时
      1. 创建一个变量保存进位标识，如果是1表示需要进位。如果是0表示不需要进位。
      2. 将数组反转。遍历数组，个、十、百千……
      3. 从个位开始判断个位是否加1后大于9。如果大于9将该数字加1后赋值给个位，进位标识赋值为1。十、百、千以此类推；如果小于9则将该数字加1后赋值给个位，进位标识赋值为0。当进位标识赋值为0时，推出循环。再次反转数组，将数组反转回原来位置。返回数组即可。
   2. 第二种情况，需要增加最高位时。则遍历完数组后临时变量仍旧为1。此时需要追加个1到数组末尾。将数组反转回原来的顺序，返回数组即可。

2. 实现

class Solution(object):
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """

        # 不需要进1
        if digits[-1] + 1 < 10:
            digits[-1] = digits[-1] + 1
            return digits

        # 需要进1，分两种情况。1不需要增加最高位，如899 变为900；2需要增加一位，如999变为1000
        digits.reverse()
        # 进位标识，真：需要，假：不需要
        plus = 1
        for i, n in enumerate(digits):
            if plus:
                if digits[i] + plus > 9:
                    digits[i] = n + plus - 10
                    plus = 1
                else:
                    digits[i] = n + plus
                    plus = 0
            else:
                break
        if plus:
            digits.append(plus)
        digits.reverse()
        return digits

3.提交