Leetcode :面试题 03.02. 栈的最小值(单栈）

最新推荐文章于 2023-04-22 17:50:50 发布

原创

最新推荐文章于 2023-04-22 17:50:50 发布 · 359 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #数据结构 #栈 #算法

设计一个栈，除了标准的push和pop操作外，还需支持min操作，返回栈中当前元素的最小值。要求所有操作的时间复杂度为O(1)。解决方案包括使用单栈或双栈，例如，单栈通过记录每次压入栈的最小值，pop时判断是否弹出的是最小值并更新最小值。

面试题 03.02. 栈的最小值

请设计一个栈，除了常规栈支持的pop与push函数以外，还支持min函数，该函数返回栈元素中的最小值。执行push、pop和min操作的时间复杂度必须为O(1)。

示例：

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

这道题双栈维护也可以做，单栈做法很巧妙。每次栈中压入更小数值，和如果去掉更小数后的最小值。如果出栈是更小数，就弹出两个，并改变最小值

class MinStack {
   
   
public:
    /** initialize your data structure here. */
    stack<int>_stack;
    int _min=INT_MAX;
    MinStack() {
   
   

    }
    
    void push(int x) {
   
   
        if(_min>=x){
   
   
            if(!_stack.empty()){
   
   
                _stack.push(_min)