算法图解:如何找出栈中的最小值?
作者 | 王磊
来源 | Java中文社群(ID:javacn666)
转载请联系授权(微信ID:GG_Stone)
前面我们学习了很多关于栈的知识,比如《动图演示:手撸堆栈的两种实现方法!》和《JDK 竟然是这样实现栈的?》,那么接下来我们再来刷一些关于栈的经典面试题以巩固学过的知识。
我们今天的面试题是这样的...
题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
LeetCode 地址:https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof/
思考
首先来说这道题目本身很好理解,它的实现难点在于以下两个方面:
当我们进行 pop(移除栈顶元素)操作时如果删除的是当前最小值,那么我们如何寻找下一个最小值?
要保证调用 min、push 及 pop 的时间复杂度都是 O(1)。
也就是说,在我们执行了 pop 时如果移除的栈中最小的值,那么如何寻找栈中的下一个最小元素?并且要保证操作的时间复杂度为 O(1)。这个时间复杂度制约了我们在移除了最小值之后不能通过遍历查找下一个最小值,所以这就成为了这道题的难点。
比如当我们移除以下栈顶元素值:
因为最小值就是 1,因此在移除之后最小值也被移除了,如下图所示:
那么接下来,让我们一起思考 3 分钟,想一想应该如何处理这个问题~
解题思路
其实我们可以在每次入栈时,判断当前元素是否小于最小值,如果小于则将原最小值和最新的最小值相继入栈,这样在调用 pop 时即使移除的是最小值,我们也能通过获取下一个元素得到一个新的最小值,执行流程如下所示。
操作步骤1
入栈第一个元素,因为是第一个元素,因此最小值就是此元素的值。
操作步骤2
入栈第二个元素,如下图所示:
因为入栈的元素 3 比 8 小,所以先将栈中的原最小值 8 存入栈中,再将 3 入栈。
操作步骤3
入栈第三个元素,如下图所示:因为入栈元素 5 大于 3,因此栈中的最小值不变,直接将元素 5 入栈。
操作步骤4
继续入栈,如下图所示:入栈元素 1 小于于 3,因此先将原最小值 3 入栈,再将 1 入栈,栈中的最小值更改为 1。
操作步骤5
执行出栈操作,如下图所示:元素 1 出栈,判断当前元素就是栈的最小值,因此将栈顶元素 3 设置为最小值,并移除元素 3,如下图所示:
操作步骤6
继续出栈,如下图所示:因为元素 5 不是当前最小值,因此直接出栈。
操作步骤7
继续出栈,如下图所示:因为出栈元素 3 为最小值,因此继续将最小值设置为栈顶元素 8,并将栈顶元素出栈,如下图所示:
这样就剩下最后一个元素了,最后一个元素出栈之后就成空栈了,整个流程就执行完了。
实现代码1
接下来我们将上面的思路用代码实现一下,我们用数组实现的栈来实现相关的功能,代码如下:
class MinStack {private int[] data; // 栈数据private int maxSize; // 最大长度private int top; // 栈顶指针(下标)private int min; // 最小值// 构造函数public MinStack() {// 设置默认值maxSize = 10000;data = new int[maxSize];top = -1;min = Integer.MAX_VALUE;}// 入栈(添加元素)public void push(int x) {if (min >= x) {// 遇到了更小的值,记录原最小值(入栈)data[++top] = min;min = x;}// 当前值入栈data[++top] = x;}// 出栈(移除栈顶元素)public void pop() {if (min == data[top]) {min = data[--top]; // 拿到原最小值,并(将原最小值)出栈}--top; // 出栈}// 查找栈顶元素public int top() {return data[top];}// 查询最小值public int min() {return min;}
}
上述代码在 LeetCode 的执行结果如下:
可以看出性能还是很高的,超越了 99.92% 的用户,内存消耗也不大。它的核心代码在 push
方法内,先将原最小值和最新最小值相继入栈,在 pop
出栈时判断出栈元素是否为最小值,如果是最小值则将当前最小值指向栈顶元素并将栈顶元素出栈,这样就得到了下一个新的最小值了。
实现代码2
如果我们不想使用数组的自定义栈来实现,还可以使用 Java 中自带的栈 Stack
来实现此功能,代码如下:
class MinStack {private Stack<Integer> stack = new Stack<>();private int min = Integer.MAX_VALUE;public MinStack() { }// 入栈(添加元素)public void push(int x) {if (x <= min) {// 遇到了更小的值,记录原最小值(入栈)stack.push(min);min = x;}stack.push(x);}// 出栈(移除栈顶元素)public void pop() {if (stack.pop() == min) {min = stack.pop(); // 取出原最小值}}// 查找栈顶元素public int top() {return stack.peek();}// 查询最小值public int min() {return min;}
}
上述代码在 LeetCode 的执行结果如下:
从结果可以看出,使用 Java 中自带的栈的性能不如自定义数组的栈,但代码还是通过了测试。这种实现方式的优点就是代码比较简单,可以利用了 Java 自身的 API 来完成了最小值的查找。
这种实现代码的方式(使用 Java API),在刷题或者实际面试中如果没有特殊说明是可以直接用的。
总结
本文我们通过两种方式:自定义数组栈和 Java API 中的 Stack
来实现了栈中最小值的功能,保证了在调用栈的 min、push 及 pop 方法时的时间复杂度都是 O(1)。两种实现方式的代码虽然略不相同,但实现思路都是一样的,都是在元素入栈时判断当前元素是否小于最小元素,如果小于最小元素则先将原最小值入栈,再将当前最小元素入栈,这样当调用 pop 方法时,即使移除的是最小值,只需要将下一个元素取出即为新的最小值了,这样就可以实现调用 min、push 及 pop 方法时的时间复杂度都是 O(1) 了。
最后
机智的你一定还有其他的实现答案,评论区告诉我吧~
原创不易,各位素质四联,谢啦。
往期推荐
链表反转的两种实现方法,后一种击败了100%的用户!
JDK 竟然是这样实现栈的?
动图演示:手撸堆栈的两种实现方法!
关注下方二维码,收获更多干货!
算法图解:如何找出栈中的最小值?相关推荐
- java栈最小值的实现,关于java:算法图解如何找出栈中的最小值
后面咱们学习了很多对于栈的常识,比方<动图演示:手撸堆栈的两种实现办法!>和<JDK 居然是这样实现栈的?>,那么接下来咱们再来刷一些对于栈的经典面试题以坚固学过的常识. 咱们 ...
- 【算法14】找出数组中两个只出现一次的数字
[题 目]一个整型数组中除了两个数字外,其他的数字都出现两次.请找出这两个只在数组中出现一次的数字.要求时间复杂度O(n),空间复杂度O(1). [思 路]首先我们考虑一个稍微简单点的情况:如果这个数 ...
- 小小c#算法题 - 1 - 找出数组中满足条件的两个数
把一些东西放到网上,看的时候方便一些.代码可能有bug且不是最佳解决方案,仅供参考. 题目:输入一个按升序排序的数组和一个数字,在数组中查找两个数,使得它们的和正好是输入的那个数字. 要求时间复杂度是 ...
- 滑动窗口算法思想,找出字符串中的所有字母异位词
力扣大神写的诗 一开始不理解滑动窗口是什么! 看完此诗我才明白,滑动窗口其实就是双指针的一种形式 左右指针齐头进. c#代码如下 public class Solution {public IList ...
- 算法期中1007. 怪兽训练 (找出有向图中所有的强连通分量的Kosaraju算法)
Description 贝爷的人生乐趣之一就是约战马会长. 他知道马会长喜欢和怪兽对决,于是他训练了N只怪兽,并对怪兽用0到N-1的整数进行编号. 贝爷训练怪兽的方式是让它们一对一互殴. 两只怪兽互殴 ...
- 数据结构第5章例题 若矩阵Am×n中存在某个元素aij满足:aij是第i行中的最小值且是第j列中的最大值,则称该元素为矩阵A的一个鞍点。试编写一个算法,找出A中的所有鞍点。
[例5.1] 若矩阵Am×n中存在某个元素aij满足:aij是第i行中的最小值且是第j列中的最大值,则称该元素为矩阵A的一个鞍点.试编写一个算法,找出A中的所有鞍点. 算法如下: void saddl ...
- C++找出数组中的第一个非重复整数的算法(附完整源码)
C++找出数组中的第一个非重复整数的算法 C++找出数组中的第一个非重复整数的算法完整源码(定义,实现,main函数测试) C++找出数组中的第一个非重复整数的算法完整源码(定义,实现,main函数测 ...
- 找出数组中第i小元素(时间复杂度Θ(n)--最坏情况为线性的选择算法
找出数组中第i小元素 期望时间复杂度:Θ(n) 最坏情况的时间复杂度Θ(n^2) int randomized_select(int *array,int start,int end,int inde ...
- 【408计算机考研】|【2018统考真题-41】| 给定一个含 n(n≥1)个整数的数组,请设计一个在时间上尽可能高效的算法,找出数组中未出现的最小正整数
目录 一.题目 二.解答 三.测试数据 一.题目 给定一个含 n(n≥1)个整数的数组,请设计一个在时间上尽可能高效的算 法,找出数组中未出现的最小正整数.例如,数组{-5, 3, 2, 3}中未 ...
最新文章
- linux下防火墙iptables原理及使用
- 【Groovy】循环控制 ( Number 注入函数实现循环 | times 函数 | upto 函数 | downto 函数 | step 函数 | 闭包作为最后参数可写在外面 )
- 动态内存(Dynamic Memory),微软的内存过量分配技术?
- 山东省2021年高考成绩查询平台6,山东2021年高考成绩改为6月26日前公布
- echart实现3d地图_3D飞线效果——让线“飞”起来的秘密
- node.js 报错Error: ENFILE: file table overflow...ENFILE: file table overflow, open...
- 写代码:实现用户输入用户名和密码,当用户名为seven且密码为123时,显示登录成功,否则失败,失败时允许重复输入三次。...
- 二维码 小优机器人_自动炒菜机器人亮相服贸会 可做3000多道菜品
- 谁蹭了我的WiFi?浅谈家用无线路由器攻防
- 快乐数Python解法
- 测试经理必知必会-Kanban和Scrum区别
- Aspose.Cells 单元格格式设置 如设置为数字
- MATLAB台大郭彦甫老师课程笔记:第一课:基本操作与矩阵输入
- windows服务编程
- android 实用代码
- C#模仿腾讯QQ源码下载(附效果图)_张童瑶的博客
- 幂运算(快速幂)Java实现
- DNS解析:腾讯云转战cloudfare解析记录
- 恒源云(GPUSHARE)_长尾分布的多标签文本分类平衡方法(论文学习笔记)
- SimpleCG绘图函数(4)--绘制圆
热门文章
- Problem D: 链表的基本运算(线性表)
- Netweaver里某个software component和C4C的版本
- 获取数据库名称dbName
- ajax的访问 WebService 的方法
- 返回表达式列表中最小值least(exp1,exp2,exp3,……,expn)
- 李洪强iOS开发之FMDB线程安全的用法
- 原生 js前端路由系统实现3之代码 构建工具 和 querystring功能
- Javascript 仿Flash 图片切换 及 Flash 图片切换
- zookeeper命令行操作
- 百度坐标转换API使用