402. 移掉 K 位数字

最新推荐文章于 2024-09-14 22:25:23 发布

江小北

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量114

点赞数

分类专栏：算法刷题笔记

本文链接：https://blog.csdn.net/FeNGQiHuALOVE/article/details/120637355

版权

算法

算法刷题笔记专栏收录该内容

157 篇文章 0 订阅

订阅专栏

在这里插入图片描述
单调栈

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        stack = []
        remove_num = 0
        for i in range(len(num)):
            while stack and num[i] < stack[-1] and k > remove_num:
                stack.pop()
                remove_num = remove_num + 1
            stack.append(num[i])
        
        if k > remove_num:
            stack = stack[:len(stack)+remove_num-k]
        
        while stack and stack[0] == '0':
            stack = stack[1:]
        if not stack:
            return '0'
        else:
            return ''.join(stack)