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