Leetcode 402. 移掉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。

**难度**: Medium

**标签**: 栈、 贪心算法、


# -*- coding: utf-8 -*-
# @Author  : LG

"""
执行用时:48 ms, 在所有 Python3 提交中击败了69.82% 的用户
内存消耗:13.9 MB, 在所有 Python3 提交中击败了6.73% 的用户

解题思路:
    具体实现见代码注释
"""
class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        result = []
        self.k = k
        def insert(c):  # 插入c字符
            # print(c, self.k, result)
            if result == [] or self.k < 1 or c >= result[-1]:   # 如果result为空,k<1,或者当前c大于栈顶元素, c入栈
                result.append(c)
            else:   # 否则,出栈一个字符,重新尝试插入c
                result.pop()
                self.k -= 1
                insert(c)

        for c in num:   # 一次插入num中数字
            if self.k > 0:
                insert(c)
            else:
                result.append(c)

        for i in range(self.k): # 当k>1时,如果栈不空,则继续出栈
            if result:
                result.pop()
        if result == []:
            result = ['0']
        return str(int(''.join(result)))