一、题目描述
ZJM 有一个长度为 n 的数列和一个大小为 k 的窗口, 窗口可以在数列上来回移动. 现在 ZJM 想知道在窗口从左往右滑的时候,每次窗口内数的最大值和最小值分别是多少. 例如:
数列是 [1 3 -1 -3 5 3 6 7], 其中 k 等于 3.
Input
输入有两行。第一行两个整数n和k分别表示数列的长度和滑动窗口的大小,1<=k<=n<=1000000。第二行有n个整数表示ZJM的数列。
Output
输出有两行。第一行输出滑动窗口在从左到右的每个位置时,滑动窗口中的最小值。第二行是最大值。
Sample Input
8 3
1 3 -1 -3 5 3 6 7
Sample Output
-1 -3 -3 -3 3 3
3 3 5 5 6 7
二、思路概述
1.使用了一个单调增队列和单调减队列,来找出每个滑动窗口的最大值和最小值。
2,以最小值为例,每次判断队列头的元素是否属于当前窗口,若不是,则从头出队列;判断队列尾的元素是否比要压进来的元素大,若是,则队尾删除,然后把元素压入队列。那么此时队列头的元素就是当前窗口的最小值。
三、细节
1.我刚开始的思考错了,没有考虑到队尾元素比较大的时候,要从队尾删除元素,只是一味地从队首删除,就。。。错了。
2.我还傻傻地以为窗口都是3大小,就。。。。没有利用到k值,感觉脑子瓦特了。
3.本来是使用了我以前自己写的一个队列类,但是超时,就改成了stl的deque,双向队列,就可以了,可能是我自己写的队列时间性能不太好。
4.stl的双向队列deque,头文件是< queue>
和普通队列相比,有pusu_back(),pop_back(),pop_front()等不一样的操作,这里主要是使用了它的队尾删除。
四、完整代码
#include<iostream>
#include <queue>
#include<stdio.h>
using namespace std;
struct point
{
int value;
int num;
};
deque<point> xp1;//找最大值
deque<point> xp2;//找最小值
int main()
{
//前期数据存储
int n,k;
scanf("%d%d",&n,&k);//n个数列长度和k为滑动窗口大小
point *a=new point[n];
for(int i=0;i<n;i++)
{
scanf("%d",&a[i].value);
a[i].num =i;
}
//计算最小值
for(int i=0;i<n;i++)
{
while (!xp2.empty() &&xp2.front().num ==i-k )//队列顶删除
{
xp2.pop_front();
}
while(!xp2.empty()&&xp2.back() .value>a[i].value) //从后面比较,从后面删除
{
xp2.pop_back() ;
}
xp2.push_back(a[i]);
if(i>=k-1)
printf("%d ",xp2.front().value);
}
printf("\n");
for(int i=0;i<n;i++)
{
//计算最大值
while(!xp1.empty()&&xp1.back() .value<a[i].value) //从后面比较,从后面删除
{
xp1.pop_back() ;
}
xp1.push_back(a[i]);
if(xp1.front().num ==i-k )//首节点比a[i]要小,或者首节点不属于这个窗口
{
xp1.pop_front() ;
}
if(i>=k-1)
printf("%d ",xp1.front().value);
}
return 0;
}