【算法】包含min函数的栈

Camellia@

已于 2023-02-17 15:31:15 修改

阅读量406

点赞数

分类专栏：算法文章标签：算法

于 2023-02-16 21:01:25 首次发布

本文链接：https://blog.csdn.net/qq_51143009/article/details/129066361

版权

算法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

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

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.

提示：

各函数的调用总次数不超过 20000 次

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/bao-han-minhan-shu-de-zhan-lcof

方法一：使用辅助栈（即需要使用额外的空间）

为啥用辅助栈呢，因为存放元素的栈pop值时可能会改变stack内部的最小值，所以引入辅助栈来维护这个最小值，这也是为啥辅助栈和主栈的元素个数应该一致的原因吧~~
官方解释说得是：元素 a b c d依次进入栈内，d最后进入栈的时候，a b c一直都在；同理，a b c d中只要最小值在，那么最小值会一直都在（故而需要引入辅助栈辅助）。

1.定义两个空栈a b，a用于存元素，b用于存a栈中最小的元素；
2.a b两个栈中元素的个数应当一致，所以pop的时候a b都需要pop一下。

class MinStack {
public:
    // 定义两个栈
    stack<int> a;
    stack<int> b;

    MinStack() {

    }
    
    void push(int x) {
        a.push(x);
        // 最开始的时候b没有任何值，b.empty确保第一次b可以存入值
        if(b.empty() || b.top() > x)
        {
            b.push(x);
        }
        else{
            b.push(b.top());
        }

    }
    
    void pop() {
        a.pop();
        b.pop();

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

    }
    
    int min() {
        return b.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->min();
 */