「面向 offer 学算法」笔面试大杀器 -- 单调栈
目录
前言
单调栈是一种比较简单的数据结构。虽然简单,但在某些题目中能发挥很好的作用。
最近很多大厂的笔试、面试中都出现了单调栈的题目,而还有不少小伙伴连单调栈是什么都不了解,因此老汪专门写了这篇文章,希望对你们有所帮助。
老规矩,先上一道题给大家看看单调栈能解决什么样的问题,这题是 2020 年猿辅导(K12 教育的独角兽,研发岗白菜价 40W 起步,不加班高福利,想要内推的可以私信老汪)的一道面试题。
给定一个二叉搜索树,并给出他的前序序列,要求输出中序序列,时间复杂度O(n),并给出证明。
单调栈
是什么:单调栈是一种具有单调性的栈,任何时刻从栈顶到栈底的元素都是单调增/减的。 干什么: 单调栈可以找到从 左/右
遍历第一个
比它大/小
的元素的位置。单调栈也可以将某个元素 左/右
边所有比它小/大
的元素按升/降序
输出。
怎么做:使用栈结构,在入栈的时候保持 id 和 val 的单调性即可。
翻译成大白话,凡是遇到题目中直接或间接要求查找某个元素左/右边第一个大于/小于它的元素,或者要求将某个元素左/右边所有小/大于它的元素按升/降序输出,就可以使用单调栈来解决。
具体怎么做你可能还有些迷糊,下面我们直接通过做题来加深对单调栈的理解。十分钟包教包会,当天就可以和面试官对线。
初入茅庐
给定数组 a,要求输出这样的数组 b,b[i] 是 a[i] 左边第一个比 b[i] 大的元素,若不存在则 b[i] = a[i]。
最暴力的解法,就是对每一个 a[i],遍历其左边的所有元素,这样所需的时间复杂度是 O(n^2)。如果使用单调栈,时间复杂度可以优化到 O(n)。
这是最基本、最直白的单调栈问题,所有单调栈问题都是在该问题上进行延伸、变种得来的,掌握了这个问题的解决方法,所有单调栈的问题都能迎刃而解。
由于本问题过于简单、直白,就不多做讲解,直接上代码:
public int[] solve(int[] a){
if(a == null) return null; //容错处理,越是简单的题越要注意细节
int n = a.length;
int[] b = new int[n];
Stack<Integer> stack = new Stack(); //单调栈当然要用栈结构啦
for(int i = 0; i < n; i++){
while(!stack.isEmpty() && stack.peek() < a[i]) stack.pop(); //所有比 a[i] 小的元素都出栈,保证了从栈顶到栈底元素是单调增的,并且栈顶的元素是 a[i] 左边第一个比 a[i] 大的元素
if(stack.isEmpty()) stack.push(a[i]);
b[i] = stack.peek();
}
return b;
}
这代码也是单调栈问题的基本结构,所有单调栈问题的代码都基于此进行变种、扩展。小伙伴们可以多花两分钟好好消化上面的代码。
考虑到有些小伙伴不用 Java,老汪把上述代码抽象成伪代码,这也是单调栈的基本结构。
背诵 + 套用,即可解决所有单调栈问题。
函数 solve (数组 a):
新建数组 b;
新建栈 stack;
For i From 0 To n - 1:
While 栈非空 且 栈顶元素 < a[i]:
出栈;
End While
If 栈空 Then:
a[i] 入栈;
End If
b[i] = 栈顶元素;
End For
return b;
小试牛刀
单调栈的最基本使用方式我们已经了解了,下面一起来解决文章开头提到的面试题。
给定一个二叉搜索树,并给出他的前序序列,要求输出中序序列,时间复杂度O(n),并给出证明。
最暴力的方法就是对前序序列进行排序,时间复杂度为 O(nlogn),使用单调栈可以优化到 O(n)。
思路分析:
对于二叉搜索树而言,给定其根节点 a[i],左子树里所有元素都比 a[i] 小,右子树里所有元素都比 a[i] 大。
回顾一下遍历顺序:
前序序列,先遍历根节点,再遍历左子树,最后遍历右子树;
中序序列,先遍历左子树,再遍历根节点,最后遍历右子树。
即,对于元素 a[i],以它为根节点的子树,
前序序列为,a[i]
, a[i] 的左子树序列
, a[i] 的右子树序列
后序序列为,a[i] 的左子树序列
, a[i]
,a[i] 的右子树序列
因此,当我们遍历前序序列,遇到右子树的第一个元素 a[j]
时,其左边所有元素都小于 a[j]
, 将其左边所有元素按升序输出,即可得到 a[i]
和 a[i] 的左子树
的后序序列。
对于右子树再以上述步骤迭代,即可得到完整的后序序列。
显然,这是单调栈的第二种用法:单调栈可以将某个元素左边所有比它小的元素按升序输出。
下面直接上代码:
public int[] solve(int[] pre){
if(pre == null) return null; //注意容错细节
int n = pre.length;
int[] infix = new int[n]; //中序序列
Stack<Integer> stack = new Stack();
int index = 0; // 指示中序序列的当前下标
for(int i = 0; i < n; i++){
while(!stack.isEmpty() && stack.peek() < pre[i]) infix[index++] = stack.pop(); //由于栈中元素是从栈顶到栈底单调增的,所以可以保证输出序列是单调增的
stack.push(pre[i]);
}
while(!stack.isEmpty()) infix[index++] = stack.pop();
return infix;
}
打怪升级
再来看一道题,这道题是 2020 年 9 月 6 日字节笔试的第二题。难度对标 leetcode 的 medium 级别。
给定一个长为 n 的序列 a。L[i] 表示第 i 个位置左边第一个大于 a[i] 的数的下标(从 1 开始),没有的话为 L[i] = 0。R[i] 表示第 i 个位置右边第一个大于 a[i] 的数的下标(从 1 开始),没有的话为 R[i] = 0。求 。
思路分析:一看题目,就是单调栈的第一种用法。使用两次单调栈分别得到 L[i] 和 R[i],再遍历一遍即可。时间复杂度为 O(n)。
代码如下:
public int solve(int[] a){
if(a == null) throw new RuntimeException("输入有误!"); //细节,一定要细
int n = a.length;
int[] L = new int[n], R = new int[n];
// 这里分两次 for 循环来写,熟练的话可以放在同一个 for 循环里。
Stack<Pair> stack = new Stack();
for(int i = 0; i < n; i++){
while(!stack.isEmpty() && stack.peek().val < a[i]) stack.pop();
L[i] = stack.isEmpty() ? 0 : stack.peek().id + 1; //注意题目要求下标是从 1 开始的
stack.push(new Pair(i, a[i]));
}
stack.clear();
for(int i = n - 1; i >= 0; i--){
while(!stack.isEmpty() && stack.peek().val < a[i]) stack.pop();
R[i] = stack.isEmpty() ? 0 : stack.peek().id + 1; //注意题目要求下标是从 1 开始的
stack.push(new Pair(i, a[i]));
}
// L[i] 和 R[i] 计算完毕,下面遍历一遍得到最大值即可
int ans = 0;
for(int i = 0; i < n; i++) ans = Math.max(ans, L[i] * R[i]);
return ans;
}
class Pair{
int id; // 下标
int val;
public Pair(int id, int val){
this.id = id;
this.val = val;
}
}
出师试炼
关卡 1
给定一个整数数组,你需要验证它是否是一个二叉搜索树正确的先序遍历序列。
你可以假定该序列中的数都是不相同的。
关卡 2
给定一个以字符串表示的非负整数 num,移除这个数中的 k 位数字,使得剩下的数字最小。
关卡 3
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
PS:在公众号【往西汪】后台回复关键字【单调栈】,即可获得上述关卡的过关秘籍(代码实现)。
本期单调栈问题就分享到这,下一期你想看什么内容呢?单调队列?前缀和思想?还是老汪独家刷题秘籍?又或者有其他想看的,也可以在下方评论区告诉老汪。
我是往西汪,致力于面向 offer 分享算法知识,希望你早日收获心仪 offer。我们下期再见。
「面向 offer 学算法」笔面试大杀器 -- 单调栈的更多相关文章
- 伯克利推出「看视频学动作」的AI智能体
伯克利曾经提出 DeepMimic框架,让智能体模仿参考动作片段来学习高难度技能.但这些参考片段都是经过动作捕捉合成的高度结构化数据,数据本身的获取需要很高的成本.而近日,他们又更进一步,提出了可以直 ...
- loj #6302. 「CodePlus 2018 3 月赛」寻找车位【线段树+单调队列】
考虑静态怎么做:枚举右边界,然后枚举上边界,对应的下边界一定单调不降,单调栈维护每一列从当前枚举的右边界向左最长空位的长度,这样是O(nm)的 注意到n>=m,所以m<=2000,可以枚举 ...
- 面试必备:高频算法题终章「图文解析 + 范例代码」之 矩阵 二进制 + 位运算 + LRU 合集
Attention 秋招接近尾声,我总结了 牛客.WanAndroid 上,有关笔试面经的帖子中出现的算法题,结合往年考题写了这一系列文章,所有文章均与 LeetCode 进行核对.测试.欢迎食用 本 ...
- 百度「Web 前端研发部」面试过程和常见问题 可能会采用哪些方法来面试 STAR 面试法 喜欢什么样的面试者 喜欢问的问题
http://segmentfault.com/a/1190000002498800 在他们的github上看到的,收藏一下备用.看完觉得还有很多要努力的地方. FEX 的面试过程 我们一般会有 3 ...
- 「面向打野编程」iOS多线程:CGD
「面向打野编程」iOS多线程:CGD 前言 参考网络其他文章而写,渣水平,抛砖引玉. 虽然Concurrent意思为并发,但由于队列的实际效果,以下称为并行队列. 当前iPhone的CPU核心数远小于 ...
- 字符串匹配「 KMP 算法 」
引言 众所周知,字符串无论是在 OI 中还是别的计算机领域都占有比较大的比重,今天说的就是一个关于匹配字符串的算法——「 KMP 算法 」. 0x00 KMP 算法用于解决这样的一类问题:给定一个文本 ...
- 「雕爷学编程」Arduino动手做(15)——手指侦测心跳模块
37款传感器和模块的提法,在网络上广泛流传,其实Arduino能够兼容的传感器模块肯定是不止37种的.鉴于本人手头积累了一些传感器与模块,依照实践出真知(动手试试)的理念,以学习和交流为目的,这里准备 ...
- 图解最长回文子串「Manacher 算法」,基础思路感性上的解析
问题描述: 给你一个字符串 s,找到 s 中最长的回文子串. 链接:https://leetcode-cn.com/problems/longest-palindromic-substring 「Ma ...
- C#下实现的K-Means优化[1]-「离群点检测」
资源下载 #本文PDF版下载 C#下实现的K-Means优化[1]-「离群点检测」 前言 在上一篇博文中,我和大家分享了「C # 下实现的多维基础K-MEANS聚类」的[C#下实现的基础K-MEANS ...
随机推荐
- 数电学习笔记之CMOS传输门工作原理
CMOS 传输门从结构上看是由一个PMOS和一个NMOS管组成 先简单粗略讲讲PMOS管和NMOS管导通与截止吧 首先我们MOS管有三个极,源极(S:Source).漏极(D:Drain)和栅极(G: ...
- java_Object类、日期时间类、System类、包装类
Object类 java.lang.Object 类是所有类的父类.它描述的所有方法子类都可以使用.在对象实例化的时候,最终找的父类就是Object. 如果一个类没有特别指定父类, 那么默认则继承自O ...
- 精讲RestTemplate第4篇-DELETE、PUT等请求方法使用详解
本文是精讲RestTemplate第5篇,前篇的blog访问地址如下: 精讲RestTemplate第1篇-在Spring或非Spring环境下如何使用 精讲RestTemplate第2篇-多种底层H ...
- python设计模式之享元模式
python设计模式之享元模式 由于对象创建的开销,面向对象的系统可能会面临性能问题.性能问题通常在资源受限的嵌入式系统中出现,比如智能手机和平板电脑.大型复杂系统中也可能会出现同样的问题,因为要在其 ...
- Autocad.net利用Xaml创建Ribbon界面
0.Ribbon界面 Ribbon界面设计语言最早于Office 2007出现,后来逐渐被其他Windows系统组件采用,一些专业软件也开始以Ribbon界面取代传统菜单栏设计.Ribbon设计语言显 ...
- python3 - selenium常用的参数
# 选用开发者模式,创建一个浏览器对象,可避免被检测到是selenium模拟浏览器 chrome_options = webdriver.ChromeOptions() chrome_options ...
- C# WebAPI项目,不支持HttpPut请求!!!
有点标题党了,这个异常的现象是这样的: 我有一个正在跑的项目,要对接第三方厂家的设备. 对方给的接口文档,里面有一个接口是需要我这边实现的,要求必须是PUT请求方式. 所以我在项目基础上,新增一个W ...
- git 常规操作 windows版
首先在本地建立好文件夹,然后初始化git仓库: git init 接下来在github上面克隆项目: git clone 这里写你的项目地址 然后就可以修改,删除,提交代码了 如果需要在新分支上面开 ...
- 你所不知道的Spring的@Autowired实现细节
前言 相信很多Java开发都遇到过一个面试题:Resource和Autowired的区别是什么?这个问题的答案相信基本都清楚,但是这两者在Spring中是如何实现的呢?这就要分析Spring源码才能知 ...
- You are using pip version 10.0.1, however version 20.2.2 is available.
在安装第三方库时,出现如下提示: You are using pip version 10.0.1, however version 20.2.2 is available.You should co ...