剑指Offer30包含min函数的栈

hesorchen

于 2022-01-26 20:51:45 发布

阅读量186

点赞数 1

CC 4.0 BY-SA版权

分类专栏： # 剑指Offer 文章标签： leetcode 链表算法

本文链接：https://blog.youkuaiyun.com/hesorchen/article/details/122708532

剑指Offer 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种特殊栈的设计方案，该栈能在O(1)时间内获取最小元素。通过使用辅助栈记录每次压栈时的最小值，实现了快速查找功能。

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

题目

剑指 Offer 30. 包含min函数的栈

为栈新增功能， $O (1)$ 得到栈中最小值。

解题思路

如果当前压入的元素不是最小值，那么它一定不会作为最小值。所以我们只需要开一个辅助栈维护好每个“在压入时是最小值”的元素。

代码

class MinStack
{

public:
    stack<int> st, st2;

    MinStack()
    {
        while (!st.empty())
            st.pop();
        while (!st2.empty())
            st2.pop();
    }

    void push(int x)
    {
        st.push(x);
        if (!st2.size() || x <= st2.top())
            st2.push(x);
    }

    void pop()
    {
        if (st2.size() && st.top() == st2.top())
            st2.pop();
        st.pop();
    }

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

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