https://www.luogu.com.cn/problem/P1886 题目链接
有一个长为 n 的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:
The array is [1,3,-1,-3,5,3,6,7], and k = 3。
观察样例可知,窗口固定不变,而当窗口滑到后面时前面的数据对答案再也没有作用了。
故此我们可以联想建造一个数据结构, 其中元素个数<=k,并且用双指针控制元素的进出,并且里面的元素易于判断大小(即单调的),由此,单调队列就出现了
对于求min:我们可以维护一个队列,里面的数据递增,当读入一个新数据时,如果比队尾的数大直接读入,否则需要不断弹出队尾直到新数据大于等于队尾,同时判断队首元素是否在这个滑动窗口里,求max反之
上代码!
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int N = 1e6 + 5;
int n, k;
int a[N], q1[N], q2[N], p1[N], p2[N];
void getmin()
{
int hed = 1, tal = 0;
for(int i = 1; i <= n; i++)
{
while(hed <= tal && q1[tal] >= a[i])
tal--;
q1[++tal] = a[i];
p1[tal] = i;
if(i - p1[hed] >= k)
hed++;
if(i >= k)
printf("%d ", q1[hed]);
}
}
void getmax()
{
int hed = 1, tal = 0;
for(int i = 1; i <= n; i++)
{
while(hed <= tal && q2[tal] <= a[i])
tal--;
q2[++tal] = a[i];
p2[tal] = i;
if(i - p2[hed] >= k)
hed++;
if(i >= k)
printf("%d ", q2[hed]);
}
}
int main()
{
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i++)
scanf("%d", &a[i]);
getmin();
printf("\n");
getmax();
return 0;
}