单调队列_LeetCode1673

该博客介绍了如何使用单调队列解决LeetCode中的1673题,即找到给定数组中最具有竞争力的长度为k的子序列。博主详细讲解了通过排除无法成为最小值的数来维护单调队列的方法,并以示例说明了过程,最终实现了O(n)的时间复杂度解决方案。
摘要由CSDN通过智能技术生成

题目描述

  输入一个整数数组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 ≤ \leq nums.length ≤ \leq 1 0 5 10^5 105
0 ≤ \leq nums[i] ≤ \leq 1 0 9 10^9 109
1 ≤ \leq k ≤ \leq nums.length

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


解题思路

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

使用单调队列:遍历一遍原数组,同时维护单调队列,单调队列首元素为当前所需区间的最小值。具体做法是:每遍历一个新数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 ) O(n) 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);
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值