leetcode 155 —— Min Stack

最新推荐文章于 2021-12-25 15:25:06 发布

原创最新推荐文章于 2021-12-25 15:25:06 发布 · 284 阅读

0 ·

CC 4.0 BY-SA版权

算法学习同时被 2 个专栏收录

166 篇文章

订阅专栏

数据结构

163 篇文章

订阅专栏

本文介绍了一种支持push、pop、top及获取最小元素操作的时间复杂度均为O(1)的数据结构——最小值栈。通过维护两个栈，一个用于存储所有元素，另一个用于保存递减序列并确保栈顶始终是最小值，以此实现高效操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

push(x) -- Push element x onto stack.
pop() -- Removes the element on top of the stack.
top() -- Get the top element.
getMin() -- Retrieve the minimum element in the stack.

思路：主要是在pop的时候可能pop出最小的值，所以用一个栈保存递减序列，栈顶为最小值。当pop的值大于min的时候，无需将min栈出栈。

class MinStack {
public:
    stack<int> stk;
    stack<int> min;
    
    void push(int x) {
        stk.push(x);
        if(min.empty() || x <= min.top())
            min.push(x);
    }

    void pop() {
        if(stk.top() == min.top())
        {
            stk.pop();
            min.pop();
        }
        else
            stk.pop();
    }

    int top() {
        return stk.top();
    }

    int getMin() {
        return min.top();
    }
};