LeetCode 每日一题402：移掉K位数字

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。

解题思路：

• 假设 k = 1，即只删除一位数字。根据数字的「位数越高 影响越大」，应该从左往右遍历，找到第一处 num[ j ] > num[ j - 1 ]，删除 j 即为最小数字。如果没有找到 num[ j ] > num[ j - 1 ]，则该数字为单调递增，删除末尾元素即为最小数字。
• 对于 k = n， 执行 n 次上述操作。（贪心算法，这里每次操作的最优解合起来即为整体最优解）

优化：

1. 维护一个单调递增栈，从左往右遍历 num，将num[ i ]放入栈顶。
2. 每次放入前，判断栈顶元素是否大于 num[ i ]，若大于则弹出栈顶元素，然后重复此操作。

方法一：单调递增栈

public String removeKdigits(String num, int k) {
    // 用双端队列维护单调递增栈，方便后面取值
    Deque<Character> deque = new LinkedList<>();

    // 遍历 num，将数字放入栈
    for (int i = 0; i < num.length(); i++) {
    	// 若栈顶元素大于当前元素，则删除栈顶元素
        while (!deque.isEmpty() && k > 0 && deque.peekLast() > num.charAt(i)) {
            deque.pollLast();
            k--;
        }
        deque.addLast(num.charAt(i));
    }

    // 若没有删除完，则数字为单调递增，删除末尾元素即可
    while (k > 0) {
        deque.pollLast();
        k--;
    }

    StringBuffer ans = new StringBuffer();
    // 去掉开始的 0
    while (!deque.isEmpty() && deque.peekFirst() == '0') {
        deque.pollFirst();
    }
    // 拼接字符串
    while (!deque.isEmpty()) {
        ans.append(deque.pollFirst());
    }
    return ans.length() == 0 ? "0" : ans.toString();
}

执行结果：