402. 移掉 K 位数字

Difficulty
Medium
Tags
单调栈
URL
https://leetcode-cn.com/problems/remove-k-digits/
Star
给你一个以字符串表示的非负整数 num 和一个整数 k ，移除这个数中的 k 位数字，使得剩下的数字最小。请你以字符串形式返回这个最小的数字。
示例 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 <= k <= num.length <= 105
  • num 仅由若干位数字（0 - 9）组成
  • 除了 0 本身之外，num 不含任何前导零

单调栈

思路
限制了出栈的次数为k，那么只需要在弹出栈的循环条件加一下该限制即可。
最后去除首位为0 的字符，采用 str(int(s))的方式。即先转为数字，再将数字转为字符串
题解
class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        if len(num) <= k:
            return '0'
        stack = []
        for i in range(0, len(num)):
            while stack and k > 0 and stack[-1] > num[i]:
                k -= 1
                stack.pop()
            stack.append(num[i])

        while k > 0 and stack:
            k -= 1
            stack.pop()
        
        
        return str(int("".join(stack)))