题目描述
现在有一堆数字共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、维护队首(对于上题就是如果队首已经是当前元素的m个之前,则队首就应该被删了,head++)
2、在队尾插入(每插入一个就要从队尾开始往前去除冗杂状态,保持单调性)
单调队列模板:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
struct node
{
int x,y;
}v[1010000]; //x表示值,y表示位置 可以理解为下标
int a[1010000],n,m,mx[1010000],mn[1010000];
void getmin()
{
int i,head=1,tail=0;// 默认起始位置为1 因为插入是v[++tail]故初始化为0
for(i=1;i<m;i++)
{
while(head<=tail && v[tail].x>=a[i]) tail--;
v[++tail].x=a[i],v[tail].y=i;
// 根据题目 前m-1个先直接进入队列
}
for(;i<=n;i++)
{
while(head<=tail && v[tail].x>=a[i]) tail--;
v[++tail].x=a[i],v[tail].y=i;
while(v[head].y<i-m+1) head++;
mn[i-m+1]=v[head].x;
// 道理同上,当然了 要把已经超出范围的从head开始排出
// 然后每个队首则是目前m个数的最小值
}
}
void getmax() //维护递减队列,超过目标数head++,得到的肯定是最大的
{ //遇到比前一个小的数tail++ ,往后排队
//遇到比之前大的数tail-- , 覆盖了当前的值。
int i,head=1,tail=0;
for(i=1;i<m;i++)
{
while(head<=tail && v[tail].x<=a[i]) tail--;
v[++tail].x=a[i],v[tail].y=i;
}
for(;i<=n;i++)
{
while(head<=tail && v[tail].x<=a[i]) tail--;
v[++tail].x=a[i],v[tail].y=i;
while(v[head].y<i-m+1) head++;
mx[i-m+1]=v[head].x;
}
}
int main()
{
int i,j;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)scanf("%d",&a[i]);
getmin();
getmax();
for(i=1;i<=n-m+1;i++)
{
if(i==1)printf("%d",mn[i]);
else printf(" %d",mn[i]);
}
printf("\n");
for(i=1;i<=n-m+1;i++)
{
if(i==1)printf("%d",mx[i]);
else printf(" %d",mx[i]);
}
printf("\n");
return 0;
}
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+8;
struct node
{
int x , y;
}op[maxn];
int a[maxn] , mn[maxn] , mx[maxn] , n , m;
void getmax()
{
int head = 1, tail = 0 , i;
for(i = 1 ; i < m ; i++)
{
while(head <= tail && op[tail].x <= a[i])tail--;
op[++tail].x = a[i] ; op[tail].y = i;
}
for(; i <= n ; i++)
{
while(head <= tail && op[tail].x <= a[i])tail--;
op[++tail].x = a[i]; op[tail].y = i;
while(op[head].y < i-m+1)head++;
mx[i-m+1] = op[head].x;
}
}
void getmin()
{
int head = 1, tail = 0 , i;
for(i = 1 ; i < m ; i++)
{
while(head <= tail && op[tail].x >= a[i])tail--;
op[++tail].x = a[i] ; op[tail].y = i;
}
for(; i <= n ; i++)
{
while(head <= tail && op[tail].x >= a[i])tail--;
op[++tail].x = a[i]; op[tail].y = i;
while(op[head].y < i-m+1)head++;
mn[i-m+1] = op[head].x;
}
}
int main()
{
cin >> n >> m;
for(int i = 1 ; i <= n ; i++)
{
scanf("%d" , &a[i]);
}
getmax();
getmin();
for(int i = 1 ; i <= n-m+1 ; i++)
{
if(i == 1)printf("%d" , mn[i]);
else printf(" %d" , mn[i]);
}
printf("\n");
for(int i = 1 ; i <= n-m+1 ; i++)
{
if(i == 1) printf("%d" , mx[i]);
else printf(" %d" , mx[i]);
}
printf("\n");
return 0;
}