题目描述
给你一个长度为N的数组,一个长为K的滑动的窗体从最左移至最右端,你只能见到窗口的K个数,每次窗体向右移动一位,如下表:
你的任务是找出窗口在各位置时的max value,min value.
数据范围:
20%: n<=500; 50%: n<=100000;
100%: n<=1000000;
输入
第1行n,k,第2行为长度为n的数组
输出
2行,第1行每个位置的min value,第2行每个位置的max value
样例输入
8 3
1 3 -1 -3 5 3 6 7
样例输出
-1 -3 -3 -3 3 3
3 3 5 5 6 7
思路:单调队列。
在处理最小值时将最小值放在队首,每当要有一个新的数字a入队,就将队尾所有比a大的数弹出,因为在后续的窗体中已经有a作为最小值,已经不再需要这些数字,这些数字就可以全部弹出。如果队首的位置与a的位置距离超过了k,则代表队首的数字已经超出了框定的范围,已经不能再作为一个最小值使用了,所以要将其弹出。当要取最小值时,只需取队首元素即可。
最大值处理方法与最小值基本相同。
代码:
#include<bits/stdc++.h>
using namespace std;
struct dd{
int u,v;
}a[1000005];
int n,k;
deque<dd>maxx,minn;
int main()
{
cin>>n>>k;
//输入需要处理的数据
for(int i = 1; i <= n; i++)
cin>>a[i].v,a[i].u=i;
//最小值
//先将前k-1,即不需要输出最小值的部分全部输入
for(int i = 1; i < k; i++)
{
if(!minn.empty())//防止.back()函数报错
{
//把所有比a[i]大的数字弹出
while(a[i].v<=minn.back().v)
{
minn.pop_back();
if(minn.empty())
break;
}
}
//让a[i]入列
minn.push_back(a[i]);
}
for(int i = k; i <= n; i++)
{
if(!minn.empty())//防止.back()函数报错
{
//同上
while(a[i].v<=minn.back().v)
{
minn.pop_back();
if(minn.empty())
break;
}
}
if(!minn.empty())//防止a[i]是最大值的情况
{
//如果队首的队列已经出了k的范围,即已经出了框定的窗体,则弹出
if(a[i].u-minn.front().u>=k)
minn.pop_front();
}
minn.push_back(a[i]);
cout<<minn.front().v<<" ";
}
cout<<endl;
//最大值同最小值,只需将判断大小符号更改方向即可
for(int i = 1; i < k; i++)
{
if(!maxx.empty())//防止.back()函数报错
{
while(a[i].v>=maxx.back().v)
{
maxx.pop_back();
if(maxx.empty())
break;
}
}
maxx.push_back(a[i]);
}
for(int i = k; i <= n; i++)
{
if(!maxx.empty())//防止.back()函数报错
{
while(a[i].v>=maxx.back().v)
{
maxx.pop_back();
if(maxx.empty())
break;
}
}
if(!maxx.empty())//防止a[i]是最大值的情况
{
if(a[i].u-maxx.front().u>=k)
maxx.pop_front();
}
maxx.push_back(a[i]);
cout<<maxx.front().v<<" ";
}
cout<<endl;
}