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();
    }
}
经验分享 程序员 微信小程序 职场和发展