leetcode-66-加1

题目

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

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

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

示例 1:

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

解法

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        ind = 0
        for i in range(1, len(digits)+1):
            if digits[-i] == 9:
                ind = -i
            else:
                break
        if ind == 0:
            digits[-1] +=  1
            return digits
        elif ind == -len(digits):
            a = [0] * len(digits)
            a.insert(0, 1)
            return a
        else:
            a = digits[:ind]
            a[-1] += 1
            a = a + [0] * abs(ind)
            return a

解题思路：

首先分析出这是个什么类型的题目，需要利用什么方法。我会想到先将加1分为以下三种情况：

• digits全为9；
• digits尾部有连续的9，但是不全为9。例如1699；
• digits个位数不为9，例如1931.
  这其中9的位置至关重要，因此可以通过指针定位连续9的最高位。然后针对不同情况，进行计算。

链接：https://leetcode-cn.com/problems/plus-one