leetcode中等402移掉k位数字

Posted 2021-06-06 qq_40707462

思路：单调栈+贪心

移除一位数字原则：①尽可能靠左②尽可能大
操作：找出第一处发生递减的地方，删除前一个数，一直递增的 话就删除最后一个数
如：1231，删除3
重复k次，时间复杂度最差会达到 O(nk)

优化：用单调栈维护，栈中的元素代表截止到当前位置，删除不超过 k 个数字后，所能得到的最小整数

操作：

• 对于每个数字，如果该数字小于栈顶元素，我们就不断地弹出（删除）栈顶元素，直到①栈为空②新的栈顶元素不大于当前数字③我们已经删除了 k位数字。
• 如果该数字大于栈顶，表示此时仍是递增，入栈即可
• 此时栈底到栈顶是单调递增的，如果弹出了不到k个元素，那再从末尾删除几个，最后记得删除最前面的0，如果没剩下的数了就返回0

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        stack=[]
        for n in num:
            while stack and k and n<stack[-1]:#维护栈内递增（不减）
                stack.pop()
                k-=1
            stack.append(n)
        if k>0:#删除的不够k个
            stack=stack[:-k]#留下从0到倒数第k个数
        return ''.join(stack).lstrip('0') or "0"#删除前置0
        #若 "".join(finalStack).lstrip('0') 这一句结果有值（真）
        #则返回该结果
        #若计算结果为空字符 ''，为假，返回 or 后面的结果 '0'