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

code

 1 #include<cstdio>
 2 #include<cstring>
 3 
 4 int q[1000100];
 5 int t[1000100];
 6 int v[1000100];
 7 int n,m,head,tail;
 8 void work_min()
 9 {
10     head = 1;
11     tail = 0;
12     for (int i=1; i<=n; ++i)
13     {
14         while (head<=tail&&q[tail]>=t[i]) tail--;
15         q[++tail] = t[i];
16         v[tail] = i;
17         while (v[head]<=i-m) head++;
18         if (i>=m) printf("%d ",q[head]);
19     }
20     printf("
");
21 }
22 void work_max()
23 {
24     memset(q,0,sizeof(q));
25     memset(v,0,sizeof(v));
26     head = 1;
27     tail = 0;
28     for (int i=1; i<=n; ++i)
29     {
30         while (head<=tail&&q[tail]<=t[i]) tail--;
31         q[++tail] = t[i];
32         v[tail] = i;
33         while (v[head]<=i-m) head++;
34         if (i>=m) printf("%d ",q[head]);
35     }
36     printf("
");
37 }
38 int main()
39 {
40     scanf("%d%d",&n,&m);
41     for (int i=1 ;i<=n; ++i)
42         scanf("%d",&t[i]);
43     work_min();
44     work_max(); 
45     return 0;
46 }