0单调栈/字符串中等 LeetCode402. 移掉 K 位数字 NC219 移掉 K 位数字

402. 移掉 K 位数字

描述

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

分析

要想去除k位之后数最小，应当让左侧的高位数尽可能的小，也可以理解为最终的数从左向右前几位应当是递增的，

• 从左向右遍历，借助单调栈，遇到比栈顶元素小的数，弹出栈顶元素，k–，如此循环直到当前元素不再小于栈顶元素或栈为空或k==0。
• 也有可能遍历结束了，k仍然不为0，说明从栈弹出的元素的个数不够(没有剔除k个数)，因此需要从栈顶弹出元素，直到k为0。

class Solution {
    public String removeKdigits(String num, int k) {
        Stack<Integer> stack = new Stack<>();
        for(int i = 0; i < num.length(); i++){
            int tmp = num.charAt(i) - '0';
            while(k > 0 && !stack.isEmpty() && tmp < stack.peek()){
                stack.pop();
                k--;
            }
            if(tmp != 0 || !stack.isEmpty()){
                stack.push(tmp);
            }
        }
        while(k > 0 && stack.size() > 0){
            stack.pop();
            k--;
        }
        if(stack.size() == 0){
            return "0";
        }
        StringBuffer sb = new StringBuffer("");
        while(!stack.isEmpty()){
            sb.insert(0,stack.pop());
        }
        return sb.toString();
    }
}