LeetCode-402. 移掉 K 位数字

题目来源

402. 移掉 K 位数字

题目详情

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

示例 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 。

提示：

• 1 <= k <= num.length <= 105
• num 仅由若干位数字（0 - 9）组成
• 除了 0 本身之外，num 不含任何前导零

题解分析

解法一：单调栈

1. 本题的一个核心问题就是如何判断一个字符是否应该被丢弃。
2. 对于两个相同长度的字符串，要判断哪个字符串转换为整数后的数值更大，只需要比较最左边不同的字符的大小。如，1a3456和1b3456。
3. 因此，我们这里可以借助一个单调栈来保存符合最优值（即删除某些字符后）的字符串，在遍历到一个字符时，不断判断栈顶的元素是否大于当前元素，如果大于则出栈。
4. 需要注意的一个点就是，在得到结果后，我们还需要删除字符串中的前导零，并且需要判断最后的字符串是否为空字符串，如果是，还需要返回“0”。

class Solution {
    public String removeKdigits(String num, int k) {
        Deque<Integer> sta = new LinkedList<>();
        int n = num.length();
        for(int i=0; i<n; i++){
            int number = num.charAt(i) - '0';
            while(!sta.isEmpty() && sta.peekFirst() > number && k > 0){
                sta.pollFirst();
                k -= 1;
            }
            sta.offerFirst(number);
        }
        while(!sta.isEmpty() && k > 0){
            sta.pollFirst();
            k -= 1;
        }

        StringBuilder sb = new StringBuilder();
        while(!sta.isEmpty()){
            sb.append(sta.pollLast());
        }
        String res = sb.toString();

        int inx = 0;
        for(int i=0; i<res.length(); i++){
            if(res.charAt(i) == '0'){
                inx++;
            }else{
                break;
            }
        }
        res = res.substring(inx, res.length());
        if("".equals(res)){
            return "0";
        }
        return res;
    }
}

相似题目

316. 去除重复字母 (困难)
321. 拼接最大数 (困难)
402. 移掉 K 位数字 (中等)
1081. 不同字符的最小子序列 （中等）

结果展示