LeetCode-移掉 K 位数字

144 阅读1分钟

算法记录

LeetCode 题目:

  给你一个以字符串表示的非负整数 num 和一个整数 k ,移除这个数中的 k 位数字,使得剩下的数字最小。请你以字符串形式返回这个最小的数字。


说明

一、题目

  注意输出不能有任何前导零。

二、分析

  • 什么样的字符会被删掉,那肯定是高位比较大的字符,也就是我们需要维护一个近似递增的队列,使得前一个总比后一个小,单调栈刚好符合。
  • 需要注意的点是需要删除 k 个字符,但是可能会遇见字符之间本身就存在一个较好的单调性,这时候就需要手动进行末尾字符的删除。
  • 不饿能有前导零,在输出的时候还需要将前导零进行剔除。
class Solution {
    public String removeKdigits(String num, int k) {
        if(k >= num.length()) return "0";
        StringBuilder builder = new StringBuilder();
        Deque<Character> stack = new LinkedList();
        for(char c : num.toCharArray()) {
            while(k > 0 && !stack.isEmpty() && stack.peekLast() > c) {
                stack.pollLast();
                k--;
            }
            stack.offer(c);
        } 
        for(int i = 0; i < k; i++) stack.pollLast();
        boolean flag = true;
        while(!stack.isEmpty()) {
            if(flag && stack.peek() == '0') {
                stack.pollFirst();
                continue;
            }
            flag = false;
            builder.append(stack.pollFirst());
        }
        return builder.length() == 0 ? "0" : builder.toString();
    }
}

总结

熟悉单调栈的使用方法以及字符的处理。