P1886 滑动窗口 单调队列

P1886

题目描述

现在有一堆数字共N个数字（N<=10^6），以及一个大小为k的窗口。现在这个从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最大值和最小值。

例如：

The array is [1 3 -1 -3 5 3 6 7], and k = 3.

输入输出格式

输入格式：

 

输入一共有两行，第一行为n,k。

第二行为n个数(<INT_MAX).

 

输出格式：

 

输出共两行，第一行为每次窗口滑动的最小值

第二行为每次窗口滑动的最大值

 

输入输出样例

输入样例#1： 复制

8 3
1 3 -1 -3 5 3 6 7

输出样例#1： 复制

-1 -3 -3 -3 3 3
3 3 5 5 6 7

说明

50%的数据，n<=10^5

100%的数据，n<=10^6

 

代码:

#include <iostream>
#include <queue>
using namespace std;
const int MAXN = 1e6 + 8;
int a[MAXN];
struct node{
    int t;
    int v;
};
deque<node> q1,q2;
int main(){
    ios::sync_with_stdio(false);
    int n,k;
    int tmp;
    cin >> n >> k;
    for(int i=1;i<=n;++i){
        cin >> tmp;
        a[i] = tmp;
    }
    for(int i=1;i<=n;i++){
        //min
        while(q1.size() && q1.back().v >= a[i])
            q1.pop_back();
        q1.push_back(node{i,a[i]});
        if(i - q1.front().t >= k) 
            q1.pop_front();
        if(i>=k) 
            cout << q1.front().v << (i!=n?" ":"");
    }
    cout << endl;
    for(int i=1;i<=n;i++){
        //max
        while(q2.size() && q2.back().v <= a[i])
            q2.pop_back();
        q2.push_back(node{i,a[i]});
        if(i - q2.front().t >= k) 
            q2.pop_front();
        if(i>=k) 
            cout << q2.front().v << (i!=n?" ":"");        
    }
    return 0;
}

 

#include <iostream>
#include <queue>
using namespace std;
int n,m;
const int MAXN = 1e6+1;
int height[MAXN];

void solve(int flag){
	deque<int> q;
	for(int i=0;i<n;i++){
		while(!q.empty() && height[q.back()] > height[i]){
			q.pop_back();
		}
		q.push_back(i);	
		if(i-q.front()>=m){
			q.pop_front();
		}
		if(i>=m-1){
			printf(i==n?"%d":"%d ",flag?-height[q.front()]:height[q.front()]);
		}
	}
	putchar('\n');
}

int main(){
	while(~scanf("%d %d",&n,&m)){
		for(int i=0;i<n;i++) scanf("%d",&height[i]);
		solve(0);
		for(int i=0;i<n;i++) height[i] = -height[i];
		solve(1);		
	}
	return 0;
}