本题是单调队列题解的入门,当然也可以使用RMQ 和 线段树,不过速度都没有单调队列那么快。
单调队列难点:
1 如何入列,保存数据 -- 最小单调队列的时候, 所有数都入列一次,在新的数据准备入列的时候,增加判断,如果当前数值小于队尾数值,那么队尾数值就出列。空队列的时候直接入列。
2 保存的数列是什么样的?
举例吧: 1 3 -1 -3 5 3 6 7
构建最小单调队列
第一个数值1的时候,空队列,那么入列,得到: 1
第二个数值3, 因为大于1;那么1不用出列,直接入列,得到: 1 3
第三个数值-1,因为小于3,故此3出列,-1入列,继续判断-1小于1,故此1也出列,剩下队列得到: -1
以此类推……
根据实际情况,本题需要的是K大小端中的最小和最大值,那么就可以记录上各个数值的下标,加以判断。即下标不在需要的判断范围之内的就出来,不过这次是从队列头出列。因为去掉这些下标不合要求的数值之后,队列的头数值依然是最小值,故此是当前答案。
#include<cstdio>
const int MAX_N = 1000001;
int N, K;
int arr[MAX_N];
int qu[MAX_N];
int index[MAX_N];
int ans[MAX_N];
void getMins()
{
int tail = 0, head = 1; //初始化tail<head表示为空列
for (int i = 1; i < K; i++) //初始化单调队列
{
while (tail >= head && qu[tail] >= arr[i]) tail--;
qu[++tail] = arr[i]; //记录可能的答案值
index[tail] = i; //记录额外需要判断的信息
}
for (int i = K; i <= N; i++)
{
while (tail >= head && qu[tail] >= arr[i]) tail--; //不符合条件出列
qu[++tail] = arr[i];
index[tail] = i;
while (index[head] <= i-K) head++;
ans[i-K] = qu[head]; //ans从下标0开始记录
}
}
void getMaxs()
{
int tail = 0, head = 1;
for (int i = 1; i < K; i++) //初始化单调队列
{
while (tail >= head && qu[tail] <= arr[i]) tail--;
qu[++tail] = arr[i];
index[tail] = i;
}
for (int i = K; i <= N; i++)
{
while (tail >= head && qu[tail] <= arr[i]) tail--; //不符合条件出列
qu[++tail] = arr[i];
index[tail] = i;
while (index[head] <= i-K) head++;
ans[i-K] = qu[head]; //ans从下标0开始记录
}
}
int main()
{
while (~scanf("%d %d", &N, &K))
{
for (int i = 1; i <= N; i++) scanf("%d", &arr[i]);
getMins();
for (int i = 0; i <= N-K; i++) printf("%d ", ans[i]);
putchar('\n');
getMaxs();
for (int i = 0; i <= N-K; i++) printf("%d ", ans[i]);
putchar('\n');
}
return 0;
}