洛谷1886滑动窗口(单调队列)

题目:

可以使用线段树或mulitset,时间复杂度是O(nlogn)。使用单调队列时间复杂度为O(n)

思路:

​ 用两个单调队列分别维护最大值和最小值。并记录下每个数的id,取时如果不在窗口范围则舍去。

代码:

#include<bits/stdc++.h>
#define mset(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int inf=0x3f3f3f3f;
deque<P> swmax;
deque<P> swmin;
vector<int> ansmax,ansmin;
vector<int> A(1000000+10);
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;++i) cin>>A[i];
    for(int i=1;i<=k;++i){
        //维护max windows
        while(!swmax.empty()&&A[i]>=swmax.back().second) swmax.pop_back();
        swmax.push_back({i,A[i]});

        while(!swmin.empty()&&A[i]<=swmin.back().second) swmin.pop_back();
        swmin.push_back({i,A[i]});
    }
    ansmax.push_back(swmax.front().second);
    ansmin.push_back(swmin.front().second);
    for(int i=k+1;i<=n;++i){
        while(!swmax.empty()&&A[i]>=swmax.back().second) swmax.pop_back();
        swmax.push_back({i,A[i]});

        while(!swmin.empty()&&A[i]<=swmin.back().second) swmin.pop_back();
        swmin.push_back({i,A[i]});
        int l=i-k+1;

        while(swmax.front().first<l) swmax.pop_front();
        ansmax.push_back(swmax.front().second);

        while(swmin.front().first<l) swmin.pop_front();
        ansmin.push_back(swmin.front().second);
    }
    for(int i:ansmin){
        cout<<i<<" ";
    }
    cout<<endl;
    for(int i:ansmax){
        cout<<i<<" ";
    }
    cout<<endl;
}