单调队列与Sliding Window

所谓单调队列——即元素具有单调性且同时保持着队列性质的数据结构。这个数据结构使用频率不是很高,笔者在之前也没有接触过,但是很多时候可以简化问题,对一些算法优化也大有裨益。

背景

记录这篇文章的背景,要从周末时候刷到的leetcode 上239上的问题开始谈起了。题目的名称叫Sliding Window Maximum。以下是这个问题的描述。

Given an array nums, there is a sliding window of size which is moving from the very left of the array to the very right. You can only see the numbers in the window. Each time the sliding window moves right by one position.

For example,

Given nums=[1,3,-1,-3,5,3,6,7], and k= 3.

Window position      Max

---------------              -----

[1  3  -1] -3  5  3  6  7      3

1 [3  -1  -3] 5  3  6  7      3

1  3 [-1  -3  5] 3  6  7      5

1  3  -1 [-3  5  3] 6  7      5

1  3  -1  -3 [5  3  6] 7      6

1  3  -1  -3  5 [3  6  7]      7

Therefore, return the max sliding window as[3,3,5,5,6,7].

我们分析下这个题目,滑动窗口每移动一次都要将原窗口中的元素弹出一个,压入一个,同时获取一个最大值。到这一步,我们能联想到什么样的数据结构呢?双向队列+优先级队列。如果我们使用一个优先级队列来做这个题目,毫无疑问是可以完成的。笔者一开始也是想使用优先级队列作为解题的方法。

但是一来对一个堆执行查找和随机删除并不是件很愉快的事情、二来每次执行heapify操作的时间复杂度为O(logk),总的时间代价需要O(nlogk),要知道,题目提示这个题目只需要线性时间内就可以解决。

那么问题来了,结合我们一开始的分析,如果我们需要在线性时间内解决这个问题,则我们的压入,弹出,获取最大这三件事都需要O(1)内完成。【这真的可能做到吗?存在这么厉害的数据结构嘛?】

单调队列

答案是能做到的。讨论区一个叫fentoyal的大神(600+的reputation)指出,这是一道典型的单调队列的问题。原文请戳这里

Sliding window minimum/maximum = monotonic queue. I smelled the solution just when I read the title.

这位大牛提供了一份可以直接拿来使用的MonotonicQueue的代码,供我们直接使用。我们来看看这份代码。


Monoqueue

单调队列的操作包括:

push:在队尾压入元素O(1)【这里的O(1)是由于所有的元素都入队一次,出队一次均摊意义上的O(1)】

pop:弹出队首的元素(最大元素)【O(1)】

max:返回队首的最大元素【O(1)】

这份代码使用双向队列deque作为容器存放元素。元素的类型设置为了pair——为每一个元素添加了一个计数器,记录的当前元素和上一个比它大的元素之间的元素个数(下文中我们称为count)。有点拗口(表达能力还需要训练...)。其实简单地说,就是记录在弹出这个元素之前还需要弹出多少个元素,我们不记录这些元素到底是什么(因为他们是什么并不重要,相反是一种浪费),而是记录这样的元素到底有多少个。

举个例子:对于一个队列a[i](ki),a[i+1](ki+1) 【括号中的数字表示count】a[i] >= a[i+1]

其中ki+1表示在a[i] 和a[i+1]中间有多少个元素。我们不关注这些元素,只关注这样的元素有多少个。如此想想,其实我们deque中保存的仅仅是有可能作为max被弹出的元素,因此我们的push操作的速度要比我们想象中快得多。【本身就是均摊意义上的O(1)】

笔者觉得这份实现较为优雅。拿来和大家分享下~侵删。


单调队列的应用

有了这个单调队列,我们可以直接使用它来解决我们这道Sliding Window

代码如下:


accept

我们利用单调队列,讲一个O(nlogn)的问题简化为了O(n).

所以当我们在优化DP问题,维护队列状态(最大最小)的时候,不妨尝试下单调队列


参考文献:

百度百科

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,560评论 4 361
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,104评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,297评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,869评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,275评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,563评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,833评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,543评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,245评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,512评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,011评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,359评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,006评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,062评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,825评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,590评论 2 273
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,501评论 2 268

推荐阅读更多精彩内容