题目描述:
给定一个以字符串表示的非负整数 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。
思路:(贪心)–时间O(n),空间O(n)
为使剩下的数字最小,应尽可能使高位数字最小。–> 贪心算法。
因此从高位到低位遍历,遇到数字大于后位的即删除。
用一个栈存保留下来的数字。即
- 栈顶元素大于num[i],弹栈;否则就入栈。
- 最终保留下在栈中的就是最终的结果的逆序。
特殊情况:
- k>0,即弹栈次数不足k,则舍去末位。
- 去掉一下前导0
代码:
class Solution {
public:
string removeKdigits(string num, int k) {
if(num.size()==k) return "0";
stack<char> stk;//单调栈
string res;
for(int i=0;i<num.size();++i){
while(!stk.empty() && stk.top()>num[i] && k>0){
stk.pop();
--k;
}
stk.push(num[i]);
}
//未删完
while(k>0){
stk.pop();
--k;
}
//栈合成字符串
while(!stk.empty()){
res+=stk.top();
stk.pop();
}
//去掉res的后缀0
while(!res.empty() && res.back()=='0'){
res.pop_back();
}
reverse(res.begin(),res.end());//逆置
if(res.empty()){
res="0";
}
return res;
}
};