始终维护一个单调递增栈即可,需要注意的是删除栈顶元素的时候要确保剩余的元素加上此时栈中的元素个数大于k,否则不能删除栈顶元素。栈中元素个数等于k时,当前遍历元素无需入栈,否则直接入栈,自己可以找一组数据模拟一下,加深理解体会,详见代码。
//递增栈
class Solution {
public:
vector<int> mostCompetitive(vector<int>& nums, int k) {
vector<int> inc_stk;
int len = nums.size();
for(int i = 0;i<len;++i) {
while(!inc_stk.empty() && nums[i]<inc_stk.back()&&inc_stk.size() + len - i > k) {
//要保证此时栈中的元素个数加上nums中剩余个数大于k才可以删除栈顶元素
inc_stk.pop_back();
}
if(inc_stk.size() < k) {
inc_stk.push_back(nums[i]);
}
}
return inc_stk;
}
};