【Leetcode 每日一题】402. 移掉K位数字(单调栈)

Leetcode 每日一题
题目链接: 402. 移掉K位数字
难度: 中等
解题思路: 维护一个单调栈,存储比栈顶节点还小的值。比栈顶节点大的前k个值需要抛弃。最终的结果即为答案(注意需要去掉前导0)。
题解:

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        
        stack = []

        for i in range(len(num)):
            while k > 0 and stack != [] and stack[-1] > num[i]:
                stack.pop()
                k -= 1

            stack.append(num[i])
        # print(stack)

        for i in range(k):
            stack.pop()
        
        
        # 去除前导零
        res = "".join(stack).lstrip('0')
        
        if res == "":
            return "0"
        else:
            return res

你可能感兴趣的:(Leetcode,leetcode,栈,stack,数据结构)