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

最新推荐文章于 2023-01-25 23:23:55 发布

头盔程序员

最新推荐文章于 2023-01-25 23:23:55 发布

阅读量74

点赞数

本文链接：https://blog.csdn.net/lizeheng/article/details/119619820

版权

本文介绍了如何使用两个栈来设计一个数据结构，实现在push、pop、min操作时的时间复杂度均为O(1)，以快速获取栈中的最小元素。通过辅助栈辅助主栈，解决了在不改变基本栈操作效率的前提下获取最小值的问题。

摘要由CSDN通过智能技术生成

包含min函数的栈

题目要求：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

思路

该题需要用到两个栈，一个作为主栈，一个作为辅助栈，每次元素进主栈时先判断元素是否小于辅助栈栈顶元素或是辅助栈元素为空，判断为真的话元素一同进辅助栈。代码如下

class MinStack {
    stack <int> S;
    stack <int> Min;
public:
    /** initialize your data structure here. */
    MinStack() {

    }
    
    void push(int x) {
        S.push(x);
        if(Min.empty()||x<=Min.top())
            Min.push(x);
    }
    
    void pop() {
        if(S.top()==Min.top())
        Min.pop();
        S.pop();
    }
    
    int top() {
        return S.top();
    }
    
    int min() {
        return Min.top();
    }
};