Leetcode 402. Remove K Digits | 思维 贪心

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);   
    }
    
};



猜你喜欢

转载自blog.csdn.net/u011026968/article/details/80189474