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
单调队列维护最大最小值
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<cmath> 5 using namespace std; 6 const int MAXN=10000001; 7 int read(int & n) 8 { 9 char c='.';int x=0,flag=0; 10 while(c<'0'||c>'9') 11 { 12 c=getchar(); 13 if(c=='-')flag=1; 14 } 15 while(c>='0'&&c<='9') 16 { 17 x=x*10+(c-48); 18 c=getchar(); 19 } 20 if(flag==1)n=-x; 21 else n=x; 22 } 23 int n,m; 24 int a[MAXN]; 25 int q[MAXN],p[MAXN],h=0,t=0; 26 void find_min() 27 { 28 h=1;t=0; 29 for(int i=1;i<=n;++i) 30 { 31 32 while(h<=t&&q[t]>=a[i]) 33 t--; 34 q[++t]=a[i]; 35 p[t]=i; 36 while(p[h]<=i-m) 37 h++; 38 if(i>=m) 39 printf("%d ",q[h]); 40 } 41 printf("\n"); 42 } 43 void find_max() 44 { 45 h=1;t=0; 46 memset(q,0,sizeof(q)); 47 memset(p,0,sizeof(p)); 48 for(int i=1;i<=n;++i) 49 { 50 while(h<=t&&q[t]<=a[i]) 51 t--; 52 q[++t]=a[i]; 53 p[t]=i; 54 while(p[h]<=i-m) 55 h++; 56 if(i>=m) 57 printf("%d ",q[h]); 58 } 59 printf("\n"); 60 } 61 int main() 62 { 63 read(n);read(m); 64 for(int i=1;i<=n;i++) 65 read(a[i]); 66 find_min(); 67 find_max(); 68 return 0; 69 }
作者:自为风月马前卒
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。