题目大意:字符串num代表一个非负整数,删除其中k位使得数字变得最小,输出不能出现前导零和空串。
分析:单调栈+贪心。贪心策略:从左往右删,如果左邻居>当前位,就应该删掉左邻居,这样才能使数字变最小。栈:遍历字符串,当前位小于栈顶元素时弹栈,否则入栈。可以直接用字符串来实现这个单调栈。注意:1.可能遍历结束时还没删够k位,需要直接删掉末尾几位,也就是直接截取字符串的前size-k位。2.可能删除导致出现了前导零,比如10200,因此要删掉前导零再返回。3.当k>=size,会删除所有数字,此时需要返回“0”,而不是空串。
代码:
class Solution {
public:
string removeKdigits(string num, int k) {
string ans; //用字符串实现单调栈
int m = num.size() - k; //num删除k位后的长度(ans的长度)
for (int i = 0;i < num.size();i++) {
while (k && ans.size() && ans.back() > num[i]) {
ans.pop_back();
k--;
}
ans += num[i];
}
ans.resize(m);
while (ans.size() && ans[0] == '0') ans.erase(ans.begin());
return ans.size() == 0 ? "0" : ans;
}
};