leetcode_402 移掉 K 位数字

124 阅读1分钟

要求

给你一个以字符串表示的非负整数 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 。

核心代码

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        n = len(num)
        t = n - k
        if n == k:
            return "0"
        res  = ""
        for i in range(n):
            while k and res and res[-1] > num[i]:
                res = res[:-1]
                k -= 1
            res += num[i]
            if k == 0:
                res += num[i + 1:]
                break
        res = res[:t]
        for i in range(len(res)):
            if res[i] != "0":
                return res[i:] if res[i:] else "0"
        return "0"

image.png

重点问题

解题思路:

我们使用栈来处理这个问题,遍历字符串,若num[i]大于栈顶元素,则num[i]入栈\ 若num[i]小于栈顶元素,栈顶元素依次出栈,直到栈顶元素小于num[i]或者出栈的总次数大于k。出栈的过程就是移掉数字的过程,当我们移除了k个数字之后,我能首先看遍历中是不是还有剩余数字,有拼接,没有不处理,我们的数字可能是以多个"000"开头的,我们在过滤下"0",最终输出的就是移除k个数字后的结果。