题目:移掉K位数字
给定一个以字符串表示的非负整数 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。
来源:力扣(LeetCode)第402题
分析:
维护一个单调递增栈,主要的方法很容易想到,但是在处理细节上我还是费了的时间,其实不难处理但是我想的有点复杂了。
思路:
- 维护一个单调递增栈。
- 依次遍历整个字符串,将元素放入栈,同时使单调栈严格单调递增。
- 如果已经弹出去k个元素了,那么剩下的值直接放入栈中即可,不用再删了。
- 如果还没到k个元素字符串就遍历完了,那么根据剩下还要加的元素数量,从栈顶依次出栈。
- 栈有可能是空,根据示例3,栈为空答案时
"0"
- 如果栈有值,那么将栈合并为字符串,得出答案。
激动人心的代码环节:
class Solution:
def removeKdigits(self, num: str, k: int) -> str:
stack = []
cnt = 0
n = len(num)
for i in range(n):
while cnt < k and stack and stack[-1] > num[i]:
tmp = stack.pop()
cnt += 1
if stack or num[i] != '0':
stack.append(num[i])
while stack and cnt < k:
stack.pop()
cnt += 1
ans = ''.join(stack)
return ans if ans != '' else '0'
复杂度分析:
- 时间复杂度:O(n) n为字符串的长度
- 空间复杂度:O(n) n为k的长度
总结:
- 单调栈的思想越来越熟了,但是在处理细节上还是不够熟练。
- 虽然自己第一遍做出来了,但是并没有想到最优算法,写的稍微有的啰嗦了。
- 我觉得一开始没处理好是因为我忽略了k这个值,于是关于k的细节没有想得很全面,导致最后自己测试时总是会发现一些小错误,浪费时间。