leetcode 402 移掉K位数字

描述

给定一个以字符串表示的非负整数 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。

思路

我们可以反过来思考，假设num的长度为n，我们需要移除k个数字，也就是说我们需要选择n-k个数字，使之构成的值最小。

class Solution {
public:
    string removeKdigits(string num, int k) {
        if(num.empty())return "0";
        int n=num.size();
        string res;
        int p=0;
        int flag=1;
        for(int i=0;i<n-k;i++){
            int min=INT_MAX;
            for(int j=p;j<=k+i;j++){
                if(num[j]<min){
                    min=num[j];
                    p=j+1;
                }
            }
            if(min!='0')flag=0;
            if(flag&&min=='0')continue;
            res.push_back((char)min);
            
        }
        return res.empty()?"0":res;
    }
};

另一种思路是用栈实现的，维持一个单调递增的栈

class Solution {
public:
    string removeKdigits(string num, int k) {
        string res;
        int p=k;
        for(auto a:num){
            while(!res.empty()&&p&&a<res.back())
                p--,res.pop_back();
            res.push_back(a);
        }
        int cnt=0;
        while(res[cnt]=='0')cnt++;
        res=res.substr(cnt,num.size()-k-cnt);
        return res.empty()?"0":res;
    }
};