155. Min Stack

最新推荐文章于 2020-08-03 10:03:11 发布

原创最新推荐文章于 2020-08-03 10:03:11 发布 · 101 阅读

0 ·

CC 4.0 BY-SA版权

leetcode学习专栏收录该内容

146 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现带有获取最小元素功能的数据结构的方法。通过维护两个栈，一个用于存储常规数据，另一个用于跟踪当前最小值，可以在常数时间内完成压栈、弹栈、获取栈顶元素及获取最小元素的操作。

这里写图片描述

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {
    }

    void push(int x) {
        s1.push(x);
        if(s2.empty())
            s2.push(x);
        else
            s2.push(x<s2.top()?x:s2.top());
    }

    void pop() {
        s1.pop();
        s2.pop();
    }

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

    int getMin() {
        return s2.top();
    }
private:
    stack<int> s1,s2;
};

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {
    sta=INT_MAX;
    }

    void push(int x) {
        if(x<=sta){
            s1.push(sta);
            sta=x;
        }
        s1.push(x);
    }

    void pop() {
        int t=s1.top();s1.pop();
        if(t==sta){
            sta=s1.top();
            s1.pop();
        }
    }

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

    int getMin() {
        return sta;
    }
private:
    int sta;
    stack<int> s1;
};