单调栈的应用:难度中等1673. 找出最具竞争力的子序列
给你一个整数数组 nums 和一个正整数 k ,返回长度为 k 且最具 竞争力 的 nums 子序列。
数组的子序列是从数组中删除一些元素(可能不删除元素)得到的序列。
在子序列 a 和子序列 b 第一个不相同的位置上,如果 a 中的数字小于 b 中对应的数字,那么我们称子序列 a 比子序列 b(相同长度下)更具 竞争力 。 例如,[1,3,4] 比 [1,3,5] 更具竞争力,在第一个不相同的位置,也就是最后一个位置上, 4 小于 5 。
示例 1:
输入:nums = [3,5,2,6], k = 2
输出:[2,6]
解释:在所有可能的子序列集合 {[3,5], [3,2], [3,6], [5,2], [5,6], [2,6]} 中,[2,6] 最具竞争力。
思路
使用单调栈,来获得一个单调递增的序列,具体做法是输入一个数时,将它与栈顶的数字进行比较,如果比栈顶小,就弹栈,接着比较,如果等于或者大于栈顶,就将其入栈,在这道题中,还需考虑k的大小,防止最后栈中的数字个数不足k个。
代码
class Solution {
public:
vector<int> mostCompetitive(vector<int>& nums, int k) {
int len = nums.size();
int index = 0;
vector<int> res ;
int count = 0;
for(int i =0;i<len ; i++){
while(res.size()!=0 &&nums[i] < res.back() && count <len-k ){
res.pop_back();
count++;
}
res.push_back(nums[i]);
}
while(res.size() > k){
res.pop_back();
}
return res;
}
};