题目
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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