402. 移掉K位数字 leetcode

原题链接

根据题意,我们需要移除k位数字,并且让剩下的数最小,很明显光是找最大的数是没用的,我们需要找的是高位数且这位数字比下一位数字大,说明这一位数字移除就可以让剩下数字更小.但是也有可能出现移除位数不够的情况,此时的单调栈内是单调上升的序列,所以移除前面的数字只会让剩下的数字更大,所以我们要移除后面的数字

这里的易错点：

1. 移除位数=字符串长度,此时剩下“0”而不是“”
2. 此外,我们可以留下相同的数字,因为如果下一位更大,而位数又足够的情况,这样就会导致答案错误

代码：

class Solution {
public:
    string removeKdigits(string num, int k) {
        stack<int> stk;
        vector<int> v;
        for(int i=0;i<num.size();i++){
            while(!stk.empty()&&num[stk.top()]>num[i]){
                v.push_back(stk.top());
                stk.pop();
                if(v.size()==k) break;
            }
            if(v.size()==k) break;
            stk.push(i);
        }
        while(v.size()<k){
            int t = 1;
            while(count(v.begin(),v.end(),num.size()-t)) t++;
            v.push_back(num.size()-t);
        }
        string p;
        for(int i=0;i<num.size();i++){
            if(count(v.begin(),v.end(),i)) continue;
            p+=num[i];
        }
        while(p[0]=='0'&&p.size()>1) p.erase(0,1);
        if(p=="") p+='0';
        return p;
    }
};

 

 

这里还有一个RMQ的做法.假设字符串长度为n,第一位一定在[1,n-m]中选,假设第1位在pos,那么第二位在[pos+1,n-m-1]中选,以此类推,选数就是区间查询最小值.