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).

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

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

输入输出样例

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

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

说明

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

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

code

#include<cstdio>
#include<cstring>

int q[1000100];
int t[1000100];
int v[1000100];
int n,m,head,tail;
void work_min()
{
head = 1;
tail = 0;
for (int i=1; i<=n; ++i)
{
while (head<=tail&&q[tail]>=t[i]) tail--;
q[++tail] = t[i];
v[tail] = i;
while (v[head]<=i-m) head++;
if (i>=m) printf("%d ",q[head]);
}
printf("\n");
}
void work_max()
{
memset(q,0,sizeof(q));
memset(v,0,sizeof(v));
head = 1;
tail = 0;
for (int i=1; i<=n; ++i)
{
while (head<=tail&&q[tail]<=t[i]) tail--;
q[++tail] = t[i];
v[tail] = i;
while (v[head]<=i-m) head++;
if (i>=m) printf("%d ",q[head]);
}
printf("\n");
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1 ;i<=n; ++i)
scanf("%d",&t[i]);
work_min();
work_max();
return 0;
}