单调队列_LeetCode1673

题目描述

  输入一个整数数组nums和一个正整数k，返回长度为k的子序列使得其字典序最小（最具竞争力）。

示例 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 $\le$ nums.length $\le$ $10^5$
0 $\le$ nums[i] $\le$$10^9$
1 $\le$ k $\le$ nums.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-the-most-competitive-subsequence

解题思路

容易想到：在下标区间[0,n-k]的最小值（假设下标为$i$）就是答案的第一个数，接下来在下标区间[i+1,n-k+1]的最小值就是答案的第二个数（假设下标为$i_2$），…，以此类推可以得到所有的数。但是如果遍历求解每个区间的最小值，总时间复杂度是$O(n^2)$，显然会超时。

使用单调队列：遍历一遍原数组，同时维护单调队列，单调队列首元素为当前所需区间的最小值。具体做法是：每遍历一个新数nums[i],从后往前把单调队列中比nums[i]大的数删掉（因为这些数在之后绝对不可能成为所需区间的最小值），然后把nums[i]加入单调队列。一旦确定某个区间的最小值后，就将头元素加入答案并从单调队列去掉，继续遍历即可。

拿示例2举例：nums = [2,4,3,3,5,4,9,6], k = 4
   2加入，queue = [2]
   4加入，queue = [2,4]
   删掉4，加入3，queue = [2,3]
   3加入，queue = [2,3,3]
   5加入，queue = [2,3,3,5],此时答案第一个数确定为首元素2，将2去掉之后继续遍历
   删掉5，加入4，queue = [3,3,4],答案第二个数为3，去掉3
   9加入，queue = [3,4,9],答案第三个数为3，去掉3
   删掉9，加入6，queue = [4,6],答案第四个数为4
因此答案为[2,3,3,4]。

时间复杂度：每个元素最多加入队列一次、从队列中删除一次，故时间复杂度$O(n)$。

代码

class Solution {
public:
    vector<int> mostCompetitive(vector<int>& nums, int k) {
		//单调队列
        vector<int> ans = {nums[0]};
        for(int i = 1; i < nums.size(); i++){
            //head表示当前单调队列的头
            //在i <= n-k时，需要确定的是第一个值，head=0
            //之后i每次+1，head+1
			int j = ans.size()-1, head = max(0, int(i-nums.size()+k));
            //比nums[i]大的全部去掉
            while(j >= head && nums[i] < ans[j]){
				ans.pop_back();
                j--;
            }
            ans.push_back(nums[i]);
        }
        //单调队列的前k个数就是答案
        return vector<int>(ans.begin(),ans.begin()+k);
    }
};