对第i个数,我们考虑前 i - 1个数,假设到第i个数,s[i] 和 前一个数s[i - 1], 我们有以下几种情况,来考虑是否删除s[i - 1]这个数:
- s[i - 1] > s[i]
删除s[i - 1],因为删除后,最后s[i - 1]的位置,由更小的s[i]代替,得到的答案比不删除时,是s[i - 1]本身更小 - s[i - 1] < s[i]
原理同上,我们不能删除 - s[i - 1] == s[i]
这个情况删和不删是等价的,那么我们就要交给下一个(i + 1)来解决,所以现在不能删
以上三步结束以后如果k还存在,则从后往前删除,因为最终得到的结果从左往右肯定是递增的。
最后删除前导0即可
code
class Solution {
public:
string removeKdigits(string num, int k) {
k = min(k, (int)num.size()); // .size () 返回的是 unsigned, 要转为 int
string res;
for (auto c : num) {
while (k && res.size() && c < res.back()) {
k--;
res.pop_back();
}
res.push_back(c);
}
while (k--) res.pop_back(); // 多出来的k 去掉尾巴, 注意k-- 所以k的结果为 -1
// 去掉前导0
k = 0;
while(k < res.size() && res[k] == '0') k++;
if (k == res.size()) res += '0';
return res.substr(k);
}
};