https://leetcode.com/problems/remove-k-digits/description/
刷面经的时候刷到的,想了下 贪心可做。
看前k+1位,哪个最小就选哪个,为了让这个最小的在最前,需要删除t位,删除之后,问题变成 从剩下的数字里,删除k-t个,获得最小的,于是dfs。 实际感觉有一些不好错误数据,OJ上。
比如
Input: num = "10200", k = 1 Output: "200"
这个例子 应该是102我觉得。
class Solution {
public:
string removeKdigits(string num, int k) {
string ans = "";
dfs(num, k, ans);
int ptr = 0;
while ( ptr + 1 < ans.size() && ans[ptr] == '0') ptr ++;
ans = ans.substr(ptr);
if (ans.size() == 0) return "0";
return ans;
}
void dfs(string nums, int k, string &ans) {
if (k <= 0) {
ans = ans + nums;
return;
}
if (k >= nums.size()) return ;
char selected = 127;
int pos = -1;
for (int i = 0; i < k + 1; i++) {
if (nums[i] < selected) {
selected = nums[i];
pos = i;
}
}
string ret = nums.substr(pos+1);
ans = ans + nums[pos];
k -= pos;
dfs(ret, k, ans);
}
};