《剑指Offer》包含min函数的栈

包含min函数的栈

http://www.nowcoder.com/questionTerminal/4c776177d2c04c2494f2555c9fcc1e49

双栈法的优化:压缩还原

方法一:简单的双栈法
在返回栈中的min值时,如果仅仅使用一个辅助变量min,则其值可能因为min元素被出栈而失效,常规的做法是额外添加一个同步栈(min栈),以保存记录之前所有的min值,相当于是使用了n个辅助变量,所以空间复杂度是O(n)。

但是,仅仅使用一两个辅助变量就真的不能达到目的了??其实是可以的!我们考虑使用一种类似压缩的方法,来将数据栈和辅助栈合并成一个栈,因为经过分析可以发现,其实各个元素的值与最小值是有关联的,这之间存在着冗余的信息!

一个初步的优化是将min栈中因为没有新的min值入栈而产生的重复的min值舍弃掉,这一做法可以参考一叶浮尘的博客,但是在最坏情况下,它的辅助空间复杂度也是O(n),即每次入栈的值都是更小值

方法二:压缩还原法
我们发现其实最小值min它本身就是一种冗余信息。为什么呢?因为每个元素在数值上都包含了min值,举个例子,假设入栈序列为:4、5、6、3、2、1,那么各轮次对应的min值就是:4、4、4、3、2、1,发现有:
4=4+0,5=4+1,6=4+2,3=4+(-1),2=3+(-1),1=2+(-1);各个元素在数值上已经包含了在它之前的最小值的值;
那么,我们是不是只要在数据栈中存储0、1、2、-1、-1、-1,然后再使用一个辅助变量min=1就可以了呢?
这样,根据单个辅助变量和栈中存储的值就能够推理出top值和min值了,具体规则如下:

入栈:

  • 压缩:将要入栈的元素value减去当前最小值min,得到一个差值diff,只存储该差值;
  • 更新:如果入栈的元素value比当前最小值min小,则要更新最小值:min=value;
  • 初始:第一次入栈比较特殊,因为此时的min变量并没有值,所以令:min=value;

出栈:

  • 更新:如果栈中存储的差值diff是负数,说明出栈的元素是当前最小值min,需要把min值更新为上一个最小值min = min - diff,否则,出栈的元素不是最小值,则不对min变量做任何操作;
  • 还原:如果栈中存储的差值diff是正数,说明 top = min + diff,否则,说明top元素本身是最小值 top = min;

致命的不足(2020.02.12)
非常感谢评论区里用户ThetaQing的提醒,她指出value-min的差值可能会发生溢出,比如一个是INT_MAX另一个是INT_MIN,这是一个致命的不足,这里找不到根本的解决方法,本来觉得使用python3可以缓解这个问题,后来觉得有点治标不治本,这个方法也因此变得没有应用价值了(想想阿丽亚娜5火箭爆炸事故,是吧),现在干脆就留着当反例好了。
当然,理论上这个想法还是OK的(理论与现实的差距)......同时这个故事还告诉我们:理论要联系实际,理论没有问题,可是计算机它的实际存储条件不支持咱也莫得法子......

故事之前(2019.08.25)
其实刚开始做这道题时,连引入最小栈这一做法都没想到,当时觉得很难,想了好久想到了Hash树(可以达到伪O(1),实际上应该是log级的)...

下附实现代码,代码并不复杂:

class Solution {
    stack<int> stack_;
    int top_, min_;
public:
    void push(int value) {
        if (stack_.empty()) // 第一次入栈需要额外考虑
            min_ = value;

        stack_.push(value - min_); // 存储入栈元素与最小值的差值
        if (value < min_) // 如果入栈元素比最小值要小则更新最小值
            min_ = value;
        top_ = value; //.......更新最新元素(top)的值
    }
    void pop() {
        if (!stack_.empty()) {
            // 如果出栈的是最小值(体现为存储值为负),则需要更新最小值
            if (stack_.top() < 0)
                min_ -= stack_.top();
            stack_.pop();

            if (!stack_.empty()) // 出栈需要更新 top 的值
                top_ = min_ + (stack_.top()>0 ? stack_.top() : 0);
        }
    }
    int top() {
        return top_;
    }
    int min() {
        return min_;
    }
};
全部评论
不考虑溢出,代码在这里 import java.util.Stack; import java.lang.Math; public class Solution { Stack<integer> stack = new Stack<>(); int min = Integer.MAX_VALUE; public void push(int node) { if(stack.isEmpty()) { min = node; stack.push(0); } else { stack.push(node - min); min = Math.min(node, min); } } public void pop() { int val = stack.pop(); if(val < 0) { min -= val; } else { min = Math.min(min + val, min); } } public int top() { int val = stack.peek(); if(val < 0) { return min; } else { return min + val; } } public int min() { return min; } }</integer>
点赞 回复 分享
发布于 2021-08-05 10:56
“”“假设入栈序列为:4、5、6、3、2、1, 那么各轮次对应的min值就是:4、4、4、3、2、1, 发现有:4=4+0,5=4+1,6=4+2,3=4+(-1),2=3+(-1),1=2+(-1); 各个元素在数值上已经包含了在它之前的最小值的值;”“” 这个对应关系是不是存在问题???
点赞 回复 分享
发布于 2021-08-04 22:59
栈不是先入后出吗,第一个出栈的差值若是负数,就是最小值
点赞 回复 分享
发布于 2020-02-28 22:02
第二种方法有bug,当最小值是-2147483648时,存入一个正整数就会溢出,请问您是怎么解决的?
点赞 回复 分享
发布于 2020-02-12 17:27
代码22行:直接加上stack.pop()就可以,因为这个时候top和min关系只有两种。1、top==min,此时stack.top()=0;2、top>min,此时stack.top()>0;
点赞 回复 分享
发布于 2020-02-09 21:18
还有异或写法的代码吗?想看一看
点赞 回复 分享
发布于 2020-01-08 18:14
如果差值溢出了呢?假如第一个push的是个负数
点赞 回复 分享
发布于 2019-12-10 21:58
用加减法也行,min初始值设为None
点赞 回复 分享
发布于 2019-11-10 15:44
那感觉不用异或来压缩,我用+或-都可以吧?
点赞 回复 分享
发布于 2019-10-25 20:21

相关推荐

写下这篇文章的时候,我正坐在从学校飞往北京的飞机上。就在今天,我的秋招终于算是有了结论,一共60场面试,拿到了字节百度美团等10+大厂offers,最终确认了腾讯给的机会。同时给我的这三个月,这三年以及从今天往前的所有人生做了个结。这句话写的真好,为什么这么说呢?本来挺久之前我就想写点什么,有特别多想记录的,从选择这个专业到选择这个岗位,从科研的疲惫到未来生活的期待,但总感觉这样写没个纲,乱成一团。直到我今天正式在系统中点击了三方的确认,我才突然发现这种感觉就是“不可逃避的结束”在向我走来,于是纲便有了。首先是这三个月的结果吧,或者换句话说,其实是秋招的结果。从我硕士选择了强化学习的研究方向,我就知道并不会有太多的岗位。从试错中学习,这听起来很符合人类的学习方式,但实际场景中哪来那么多试错的成本?除了游戏产业和机器人行业,我想不到特别对口的赛道,而这两个行业国内又只有寡头,让我望而生畏。整个秋招,我没法像学后端开发的同学一样投递大量的简历,我没法像学大模型的同学一样是时代的香饽饽,我只能盯着那几家公司去投,或者想方设法的在别的不太相关的算法岗上沾沾边。方向是大于努力的,但努力一定不是不重要的。秋招整体对我来说还算顺利,前文就自然变成了只有我自己懂的无病呻吟,不再赘述。从结果来说,我的秋招是非常成功的,至少我自己是满意的。命运给了我很大的惊喜,我从未想过能够在这次有多个远超期待的offer,所以我如今是心满意足。虽说很多事都是焉知非福吧,但对口的工作内容,熟悉的工作环境,我一定不会后悔。我就是这样,毕竟让我在做一百次选择也不会变,那为什么要在不可预测的未来后悔。然后是三年,三年即将过去,我的硕士生涯来到了最后一章。回想过往,我在其中反复感受井底之蛙的狭隘。从我在二十多个四点睡的凌晨产出的论文初稿开始,链式反应就这样发生了。把论文投出去,我发了一篇很长的朋友圈,那时候觉得压力真的好大,尽管其实根本没人要求我什么。那时,我第一次觉得我比本科毕业时的自己进步了太多,可以独当一面了。然后去了北京自所交流,尽管大多的时间都在修改那篇返稿的文章,但也在不一样的平台中见识了人外有人的世界。回来后,我第二次觉得自己有了很大的进步,而鄙夷去北京前的自己是如此短浅。那是11月,我开始纠结到底未来该从事开发岗还是算法岗,但时间并没有给我机会。我偷懒了,两个月根本没有做任何开发岗的准备,于是只能硬闯算法。期间只有那篇论文中了让我稍微有些自信,毕竟只有两周的理论准备时间让我心里太虚了,这甚至还算上了刷题的时间。第一面就是最想去的公司,我甚至紧张到大脑一片空白。好在后面算是有惊无险,拿到了腾讯给我的实习机会。去腾讯工作的时间是幸福的,组里氛围也很好,在公司获得的提升我觉得甚至超过了我在学校一年的量。毕竟做算法,思维的敏捷度和见识广度都是如此重要。看着同事前辈们的工作能力,和工业级的项目架构,我又一次不由得感叹曾经自己的狭隘。于是每天我只睡五小时,忙完工作忙学校,每每想到这里,我也不觉得我的成功是侥幸了。我真的建议大家离开自己舒适的环境到外面看看,鸡头或许真的不如凤尾。硕士是一个连锁反应最直接,最有力的时期。高考失利或许还能补救,考研没上岸还有第二次机会,但就业前这一年,努力就是会有回报,就一定会体现在结果中,没有侥幸。最后,也是我最想聊的。十九年的学生生涯终于快要画下句号,我其实一直觉得非常梦幻。我能回忆起每一个瞬间,有小学六年级遇到的很有个性的数学老师,有考上重点中学的快乐,有中考和提前高考而大失败的难受,有本科比赛的每个通宵的焦虑,有保研出现差错的绝望,有刚读研高压之下的崩溃。但这篇长文不会再有更多的剧情了,每个故事都让我无限回味,成为了我一生中最宝贵的财富。这些瞬间组成了我。我父亲说我是一个总抓不住机会的人,确实有很多别人没有的机会摆在我面前,我都错过了。但我心中的热爱始终没有错过,我觉得这对我来说是幸运且幸福的。我非常爱打游戏,从初中开始学编程,第一个目的就是做出属于自己的游戏,做了很多小游戏发在班级群里,被人厌烦。高中自己买了unity的书,想做自己的游戏,无奈连网络的基本知识都不懂,无功而返。到了大学,我又被强化学习吸引,我想知道能不能让人工智能来帮我打游戏呢?这一整条线我没有放弃过,拿到了游戏算法offer,我真的特别特别开心。人不是一直成功的,我经历过的失败远超过成功10倍,但那让我知道成功来之不易,让我知道失败是生活常态,让我知道真正的怯懦不是不敢失败,而是不敢尝试。言尽于此,这些都“不可逃避的结束”了。追风赶月莫停留,平芜尽处是春山。
肖先生~:追风赶月莫停留,平芜尽处是春山,passion!
我的秋招日记
点赞 评论 收藏
分享
评论
69
2
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务