单调队列以前曾看过,就是没做过题,这次做了一道有关的题,正好写下来理理思路。
poj2823
可以将它改为单调队列的经典题意,有一个k米的窗户,和N米长的数字串,每个数字占1米,问当数字串穿过窗户时,每一瞬间的最大值与最小值。讲的不太清,将样例说说吧
Window position Minimum value Maximum value
[1 3 -1] -3 5 3 6 7 -1 3
1 [3 -1 -3] 5 3 6 7 -3 3
1 3 [-1 -3 5] 3 6 7 -3 5
1 3 -1 [-3 5 3] 6 7 -3 5
1 3 -1 -3 [5 3 6] 7 3 6
1 3 -1 -3 5 [3 6 7] 3 7
上面就是样例的作答过程,窗口长度为 3 ,因此中括号每次括三个数字,理解了这道题我们再去看看单调队列。单调队列顾名思义,就是在普通队列的基础上使队列内单调(递增/递减),我一开始看时就想起了优先队列,但是发现题中的出队顺序并没有按那个递增序列来出队,简单来说,就是普通队列的进出队规则并没改变,但我们要求出每一瞬间队列中的最大值/最小值。
看看代码的表示
#include <iostream>
#include <cstdio>
using namespace std;
int N, K;
int num[660000], Q[660000], MarK[660000], M[660000];
num 数字串,Q 单调队列(数组模拟),Mark 标记此时进入窗户的最左边的下标
M 记录每一瞬间的最大/小值
void get_min()
{
int head = 1, tail = 0; 单调队列中,编码由1开始
int i;
for(i = 0; i < K-1; i++)
这是最开始传入,并没有填满窗户,所以i 是到 K-1 就结束
{
while(head <= tail && Q[tail] >= num[i]) tail--;
这句话很重要,首先head <= tail,这是在保证队列中已经有了数字,在保证了这个情况下,才有了&&后的比较,要将队尾大于等于我接下来要加入数字的所有数字都删掉,直到遇到小于num[i]的数字,或为空。(重要的是为什么这样做)当我把num[i]加进去时,num[i]就是已经在窗户中的了,那么既然我要算窗户中的最小值,那么num[i]之前,比num[i]大(或等于)的数字就没有存在的必要,反正他们都不能比num[i]小,所以就将他们删掉
Q[++tail] = num[i]; 将num[i]加入队列
MarK[tail] = i;
这时Mark[tail--]+1 到 i 之间 i 是最小值,标记Q[tail]值的下标
}
for(i = i; i < N; i++)
{
while(head <= tail && Q[tail] >= num[i])
tail--;
Q[++tail] = num[i];
MarK[tail] = i;
while(MarK[head] < i-K+1) head++;
head随着我们数字串的移动,也跟着移动地找答案,若随head的移动,Mark[head]大于了i-k+1,那么他就是i-K+1 到 i 之间的最小值,我一开始并不能很懂,但想了想我们在设置Mark时所想的(向上看至38行),那么i-k+1到i之间肯定有很多Mark[x]在其中,那么最靠前的不就是最小的,那么一旦Mark[head]大于等于了i-k+1,Q[head]就是此时的最小值
if(i-K+1 == 0) printf("%d", Q[head]);
else printf(" %d", Q[head]);
}
printf("\n");
return ;
}
void get_max() 这是求最大的,同上,没什么差
{
int head = 1, tail = 0;
int i;
for(i = 0; i < K-1; i++)
{
while(head <= tail && Q[tail] <= num[i])
tail--;
Q[++tail] = num[i];
MarK[tail] = i;
}
for(i = i; i < N; i++)
{
while(head <= tail && Q[tail] <= num[i])
tail--;
Q[++tail] = num[i];
MarK[tail] = i;
while(MarK[head] < i-K+1)
head++;
if(i-K+1 == 0) printf("%d", Q[head]);
else printf(" %d", Q[head]);
}
printf("\n");
return ;
}
int main()
{
while(~scanf("%d%d", &N, &K))
{
for(int i = 0; i < N; i++)
{
scanf("%d", &num[i]);
}
get_min();
get_max();
}
return 0;
}
若有错,请大家多多指教啦~~~