0402：移掉 K 位数字（★）

力扣第 402 题

题目

给你一个以字符串表示的非负整数 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 不含任何前导零

分析

• 类似 0316，还更简单些
• 遍历 num，假如还没删够，且当前字符比前一字符小，删掉前一字符
• 如果最终还没删够，去掉最后几位即可

解答

1
2
3
4
5
6
7
8
9

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        sk = []
        for c in num:
            while k and sk and sk[-1]>c:
                sk.pop()
                k -= 1
            sk.append(c)
        return ''.join(sk[:-k] if k else sk).lstrip('0') or '0'

56 ms