leetcode【中等】402、移掉k位数字

在这里插入图片描述

思路:单调(递增)栈+贪心

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

class Solution {
    public String removeKdigits(String num, int k) {
        if(num.length()==k) return "0";
        StringBuilder s=new StringBuilder(num);
        for(int i=0;i<k;i++){
            int temp=0;
            while(temp<s.length()-1 && s.charAt(temp)<=s.charAt(temp+1)){
                temp++;
            }          
            if(temp==s.length()-1){
                s.delete(s.length()-1,s.length());
            }
            s.delete(temp,temp+1);
        }
        //删除前置零
        while (s.length() > 1 && s.charAt(0) == '0') s.delete(0, 1);
        return s.toString();
    }
}

优化:用单调(递增)栈维护,栈中的元素代表截止到当前位置,删除不超过 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'
class Solution {
    public String removeKdigits(String num, int k) {
        if(num.length()==k) return "0";
        StringBuilder stack=new StringBuilder();
        for(char c:num.toCharArray()){
            while(stack.length()!=0 && k!=0 && c<stack.charAt(stack.length()-1)){
                stack.delete(stack.length()-1,stack.length());
                k--;
            }
            stack.append(c);
        }
        if(k!=0){
            stack.delete(stack.length()-k,stack.length());
        }
        while (stack.length() > 1 && stack.charAt(0) == '0') stack.delete(0, 1);
        return stack.toString();
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值