LeetCode 402.移掉K位数字

给你一个以字符串表示的非负整数 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位数。AC代码如下:

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        if k >= len(num):
            return "0"
        stack = []
        for i in num:
            while k and stack and stack[-1] > i:
                stack.pop()
                k -= 1
            stack.append(i)
        while k:
            stack.pop()
            k -= 1
        return str(int("".join(stack)))
经验分享 程序员 微信小程序 职场和发展