题目描述
现在有一堆数字共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
#include<iostream>
#include<cstdio>
#include<cstring>
#define N 10000000
using namespace std;
int n,k,a[N],q[N],p[N];
void MIN()
{
int head=,tail=;
for(int i=;i<k;i++)
{
while(head<=tail&&q[tail]>=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
}
for(int i=k;i<=n;i++)
{
while(head<=tail&&q[tail]>=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
while(p[head]<=i-k)
head++;
printf("%d ",q[head]);
if(i==n) printf("\n");
}
}
void MAX()
{
int head=,tail=;
for(int i=;i<k;i++)
{
while(head<=tail&&q[tail]<=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
}
for(int i=k;i<=n;i++)
{
while(head<=tail&&q[tail]<=a[i])
tail--;
q[++tail]=a[i];
p[tail]=i;
while(p[head]<=i-k)
head++;
printf("%d ",q[head]);
if(i==n) printf("\n");
}
}
int main()
{
cin>>n>>k;
for(int i=;i<=n;i++)
cin>>a[i];
MIN();
MAX(); return ;
}
思路:单调队列~~