【剑指offer】包含min函数的栈（栈）

最新推荐文章于 2020-08-02 20:56:35 发布

原创最新推荐文章于 2020-08-02 20:56:35 发布 · 69 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

80 篇文章

订阅专栏

本文介绍了一种特殊的数据结构——栈，通过使用两个栈（主栈和辅助栈），实现了在O(1)的时间复杂度内获取栈中最小元素的功能。文章详细解释了如何在压栈和弹栈操作中维护辅助栈，以确保其栈顶始终为主栈中的最小值。

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

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

代码

class Solution {
public:
    stack<int> stackMain;
    stack<int> stackMin;    // 辅助栈：栈顶元素为当前主栈内的最小值
    void push(int value) {
        if(stackMin.empty())
            stackMin.push(value);
        else if(stackMin.top() > value)
            stackMin.push(value);
        stackMain.push(value);       
    }
    void pop() {
        if(stackMain.top() == stackMin.top())    // 当弹出元素为最小值时，也弹出辅助栈栈顶元素
            stackMin.pop();
        stackMain.pop();
    }
    int top() {
        return stackMain.top();
    }
    int min() {
        return stackMin.top();
    }
};