题目描述
给定一个以字符串表示的非负整数 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。
解题思路:
使用贪心算法+栈来实现
遍历字符串,将遍历的字符压入栈中,若当前字符大于栈顶元素,则压入栈,若小于栈顶元素,则弹出栈顶元素,把当前元素压入栈中。
注意:
若当前字符为0,且栈为空,跳过。
若k>0,且栈不为空,则需要将栈弹出k个数字
最后若栈为空,则直接返回“0”
若不为空,则将栈中元素依次输出,注意要反转。
string removeKdigits(string num, int k) {
if (k >= num.size() || num.size() == 0) {
return "0";
}
string result = "";
stack<int> s;
for (int i = 0; i < num.size(); i++) {
int number = num[i] - '0';
while (!s.empty() && number < s.top() && k > 0) {
s.pop();
k--;
}
if (!s.empty() || number != 0) {//处理字符串中的0
s.push(number);
}
}
while (!s.empty() && k > 0) {//处理类似123456,k = 3这种情况
s.pop();
k--;
}
if (s.empty()) {
return "0";
}
while (!s.empty()) {
result = to_string(s.top()) + result;
s.pop();
}
return result;
}