402. 移掉K位数字

题目描述

给定一个以字符串表示的非负整数 num，移除这个数中的 k 位数字，使得剩下的数字最小。

注意:

num 的长度小于 10002 且 ≥ k。
num 不会包含任何前导零。

示例 1 :

输入: num = "1432219", k = 3
输出: "1219"
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。

示例 2 :

输入: num = "10200", k = 1
输出: "200"
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。

示例 3 :

输入: num = "10", k = 2
输出: "0"
解释: 从原数字移除所有的数字，剩余为空就是0。

解法

class Solution {
    public String removeKdigits(String num, int k) {
        //特殊处理
        if(num.length() == k) return "0";

       Stack<Character> stack = new Stack<>();

        int i=0;
        while(i<num.length()){

            //如果栈顶大于遍历的数字则弹出
            while(k > 0 && !stack.isEmpty() && num.charAt(i) < stack.peek()){
                stack.pop();
                k--;
            }
            stack.push(num.charAt(i));     
            i++;
        }

        //如果k大于0，则把栈弹出
        while(k > 0){
            stack.pop();
            k--;
        }

        //拼接
        String result = "";
        while(!stack.isEmpty()){
            result = stack.pop() + result;
        }

        //去除0开头
        while(result.startsWith("0") && result.length() > 1){
            result = result.substring(1);
        }

        return result;
    }
}