题目:移掉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题

链接:https://leetcode-cn.com/problems/remove-k-digits

分析:

维护一个单调递增栈,主要的方法很容易想到,但是在处理细节上我还是费了的时间,其实不难处理但是我想的有点复杂了。

思路:

  • 维护一个单调递增栈。
  • 依次遍历整个字符串,将元素放入栈,同时使单调栈严格单调递增。
  • 如果已经弹出去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的细节没有想得很全面,导致最后自己测试时总是会发现一些小错误,浪费时间。