问题描述
给定一个以字符串表示的非负整数 num，移除这个数中的 k 位数字，使得剩下的数字最小。
注意:
- num 的长度小于 10002 且 ≥ k。
- num 不会包含任何前导零。
示例 1 :
输入: num = "1432219", k = 3
输出: "1219"
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
示例 2 :
输入: num = "10200", k = 1
输出: "200"
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。
示例 3 :
输入: num = "10", k = 2
输出: "0"
解释: 从原数字移除所有的数字，剩余为空就是0。
解题思路
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24 | from collections import deque
class Solution:
def removeKdigits(self, num: str, k: int) -> str:
q = deque([num[0]])
for x in num[1:]:
while q and k and x < q[-1]:
q.pop()
k -= 1
q.append(x)
# 已经有序但 k 不为 0
# 则删除末尾的 k 个数
while q and k:
q.pop()
k -= 1
# 删除前面多余的 0
while q and q[0] == "0":
q.popleft()
return "".join(q) if q else "0"
|
测试数据
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 | solution = Solution()
num = "1432219"
k = 3
assert solution.removeKdigits(num, k) == "1219"
num = "10200"
k = 1
assert solution.removeKdigits(num, k) == "200"
num = "10"
k = 2
assert solution.removeKdigits(num, k) == "0"
print("PASS!")
|