LeetCode1673. 找出最具竞争力的子序列--单调栈

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] 最具竞争力。

示例 2：

输入：nums = [2,4,3,3,5,4,9,6], k = 4
输出：[2,3,3,4]

提示：

1 <= nums.length <= 105
0 <= nums[i] <= 109
1 <= k <= nums.length

题解

很有趣的题目，感觉在哪儿见过，我这样解释，因为要求k个数字，所以相当于是在这样的区间找最小的数字
一开始区间[0,n-k]，然后找到这个区间最小数字的下标为pos1,然后更新，
找第二次区间[pos1+1,n-k+1]，然后找到这个区间最小数字的下标为pos2，然后更新，
找第三次区间[pos2+1,n-k+2]，一直这样重复下去，每个区间挑出来的数字组合成的序列就是答案。

问题是数据量比较大，怎么快速定位一个区间的最小值是哪个呢？
总不可能循环跑一遍找最小吧，那绝对超时了。

于是我们利用反向思维，我们可以不用针对每个区间去找，如果我们知道每个数字在哪个区间是最小的呢？不也是可以到达这个目的吗，我们利用单调栈找出数字nums[i]在区间L[i]和R[i]是最小的，然后我们从左到右遍历，假设当前需要查询的区间为[l,r]，只需要判断当前数字nums[i]的L[i]和R[i]是否包含了[l,r]，如果包含了这个数字就是我们目标查询的最小数字。

AC代码

class Solution {
public:
    int L[100010],R[100010],arr[100010];
    stack<int>q;
    void Stack_clear()//清空栈函数 
    {
        while(!q.empty())
        q.pop();
    }
    //这里数组以下标1开始
    void get_L(vector<int>arr)//得到左区间 
    {
        for(int i=0;i<arr.size();i++)
        L[i]=i;//初始化
        Stack_clear();
        for(int i=0;i<arr.size();i++)
        {
            if(q.empty()||arr[q.top()]<arr[i])
            q.push(i);
            else
            {
                while(q.empty()==false&&arr[q.top()]>=arr[i])
                {
                    L[i]=L[q.top()];
                    q.pop();
                }
                q.push(i);
            }
        }
    }
    void get_R(vector<int>arr)//得到右区间，其实就是把数组反向遍历的左区间求解 
    {
        for(int i=0;i<arr.size();i++)
        R[i]=i;//初始化
        Stack_clear();
        for(int i=arr.size()-1;i>=0;i--)
        {
            if(q.empty()||arr[q.top()]<arr[i])
            q.push(i);
            else
            {
                while(q.empty()==false&&arr[q.top()]>=arr[i])
                {
                    R[i]=R[q.top()];
                    q.pop();
                }
                q.push(i);
            }
        }
    }
    vector<int> mostCompetitive(vector<int>& nums, int k) 
    {
        get_L(nums);
	    get_R(nums);
        vector<int>res;
        int l=0,r=nums.size()-k;
        for(int i=0;i<nums.size();i++)
        {
            if(L[i]<=l&&R[i]>=r)
            {
                res.push_back(nums[i]);
                l=i+1;
                r++;
            }
        }
        return res;
    }
};