[每日一题]87：移掉K位数字（小米OJ - 9）

题目描述

给定一个以字符串表示的非负整数 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。

题解思路

对于两个相同长度的数字序列，最左边不同的数字决定了这两个数字的大小，例如，对于 A = 1axxx，B = 1bxxx，如果 a > b 则 A > B。

基于此，我们可以知道，若要使得剩下的数字最小，需要保证靠前的数字尽可能小。

方法一：贪心 + 单调栈

我们可以用一个栈维护当前的答案序列，栈中的元素代表截止到当前位置，删除不超过 k 次个数字后，所能得到的最小整数。根据之前的讨论：在使用 k 个删除次数之前，栈中的序列从栈底到栈顶单调不降。

因此，对于每个数字，如果该数字小于栈顶元素，我们就不断地弹出栈顶元素，直到

• 栈为空
• 或者新的栈顶元素不大于当前数字
• 或者我们已经删除了 k 位数字

上述步骤结束后我们还需要针对一些情况做额外的处理：

• 如果我们删除了 m 个数字且 m < k，这种情况下我们需要从序列尾部删除额外的 k−m 个数字。
• 如果最终的数字序列存在前导零，我们要删去前导零。
• 如果最终数字序列为空，我们应该返回 0。

最终，从栈底到栈顶的答案序列即为最小数。

考虑到栈的特点是后进先出，如果通过栈实现，则需要将栈内元素依次弹出然后进行翻转才能得到最小数。为了避免翻转操作，可以使用双端队列代替栈的实现。

复杂度分析：

• 时间复杂度：$O(n)$，其中 n 为字符串的长度。尽管存在嵌套循环，但内部循环最多运行 k 次。由于 0 < k ≤ n，主循环的时间复杂度被限制在 2n 以内。对于主循环之外的逻辑，它们的时间复杂度是 $O(n)$，因此总时间复杂度为 $O(n)$。
• 空间复杂度：$O(n)$。栈存储数字需要线性的空间。

代码实现：

#include <iostream>
#include <vector>
#include <string>
using namespace std;

string removeKdigits(string num, int k) {
    if (num.size() == k) return "0";

    vector<char> stk;    //单调递增栈
    for (auto digit : num) {
        while (stk.size() > 0 && stk.back() > digit && k > 0) {
            stk.pop_back();
            --k;
        }
        stk.push_back(digit);
    }

    while (k > 0) {
        stk.pop_back();
        --k;
    }

    string res = "";
    bool flag = true;   // 用来判定是否为前置0
    for (auto digit : stk) {
        if (flag && digit == '0') {
            continue;
        }

        flag = false;
        res += digit;
    }

    return res == "" ? "0" : res;
}

int main() {
    string num;
    int k;

    while (cin >> num >> k) {
        cout << removeKdigits(num, k) << endl;
    }

    return 0;
}

方法二：用string实现的单调栈

代码示例：

#include <iostream>
#include <string>
using namespace std;

string removeKdigits(string num, int k) {
	string res;
	int n = num.size(), m = n - k;
	for (char c : num) {
		while (k && res.size() && res.back() > c) {
			res.pop_back();
			--k;
		}
		res.push_back(c);
	}
	res.resize(m);
	//去除前导0， 如10200，k = 1
	while (!res.empty() && res[0] == '0') {
		res.erase(res.begin());
	}
	return res.empty() ? "0" : res;
}

int main() {
	string s;
	int K;
	while (cin >> s >> K) {
		cout << removeKdigits(s, K) << endl;
	}
	return 0;
}