实现一个含时间复杂度为O(1)的min函数的栈

包含min函数的栈

第二天打卡,废话不多说,我们开始吧。
题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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.

提示:
各函数的调用总次数不超过 20000 次

思路:

我们通过读题,可以发现需要在常量级的时间内找到最小值!
那自然就可以想到,我们绝不能在需要最小值的时候,再做排序,查找等操作来获取!所以,我们可以创建两个栈,一个栈是原栈 min,正常进行操作,而辅助栈assi则存放应主栈不同时期的最小值。

代码实现

class MinStack {
    stack<int>  mi,ass;
public:
    /** initialize your data structure here. */
    MinStack() {
        ass.push(INT_MAX); //初始化辅助栈,使top函数能正常操作
        }
  
    void push(int x) {      //压栈的时候我们对原栈和辅助栈都要进行操作,
       //原栈就正常压,辅助栈在压的时候进行判断,压当前栈和输入中的最小的
            mi.push(x);
            if(x<ass.top()){
                ass.push(x);
            }
            else ass.push(ass.top());
    }
    
    void pop() {   //弹出的时候两者一致,直接弹,这是因为我们前面在压的时候保持两个栈都压了。
            mi.pop();
            ass.pop();   
              }
    
    int top() {
         return mi.top();
    }                        //返回原栈的当前栈顶元素
    
    int min() {
        int res = ass.top();
       return res;
    }                         //返回辅助栈当前栈顶元素
};

运行结果:

![在这里插入图片描述](https://img-blog.csdnimg.cn/1c38e8a9b4844bee9a35d1b767fa67e0.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA57u057qz5pav6auY5aSa5bCR,size_18,color_FFFFFF,t_70,g_se,x_16在这里插入图片描述

时间复杂度:其实题目中就已经给出要求,这些操作都在常数级时间复杂度。
空间复杂度:在这里我们用了两个栈来储存数据,所以空间复杂度是O(n)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值