题目描述
给定一个以字符串表示的非负整数 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 ) O(n) O(n),其中 n 为字符串的长度。尽管存在嵌套循环,但内部循环最多运行 k 次。由于 0 < k ≤ n,主循环的时间复杂度被限制在 2n 以内。对于主循环之外的逻辑,它们的时间复杂度是 O ( n ) O(n) O(n),因此总时间复杂度为 O ( n ) O(n) O(n)。
- 空间复杂度: 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;
}