【leetcode】155. 最小栈（intersection-of-two-linked-lists）（模拟）[简单]

链接

https://leetcode-cn.com/problems/min-stack/

耗时

解题：5 min
题解：15 min

题意

设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。

• push(x) —— 将元素 x 推入栈中。
• pop() —— 删除栈顶的元素。
• top() —— 获取栈顶元素。
• getMin() —— 检索栈中的最小元素。

提示：

pop、top 和 getMin 操作总是在 非空栈 上调用。

思路

设置两个栈，一个用来放元素，一个用来放当前最小元素。具体来说，最小元素栈的栈顶元素即为当前栈中最小元素。
当 push 时，如果推入的元素 小于等于 最小元素栈的栈顶元素，则也将新元素推入最小元素栈；
当 pop 时，如果弹出的的元素 等于 最小元素栈的栈顶元素，那么也弹出最小元素栈的栈顶元素。

时间复杂度：$O(1)$

AC代码

class MinStack {
private:
    stack<int> data, minst;
    
public:
    /** initialize your data structure here. */
    MinStack() {
        
    }
    
    void push(int x) {
        data.push(x);
        if(minst.empty() || x <= minst.top()) {
            minst.push(x);
        }
    }
    
    void pop() {
        if(data.top() == minst.top()) {
            minst.pop();
        }
        data.pop();
    }
    
    int top() {
        return data.top();
    }
    
    int getMin() {
        return minst.top();
    }
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack* obj = new MinStack();
 * obj->push(x);
 * obj->pop();
 * int param_3 = obj->top();
 * int param_4 = obj->getMin();
 */