[LeetCode] Min Stack 栈

#include <iostream>
#include <stack>
using namespace std;

class MinStack {
public:
    stack<int> data;
    stack<int> support;

    void push(int x) {
        if(data.empty()){
            data.push(x);
            support.push(x);
        }
        else{
            data.push(x);
            if(x<=support.top()) support.push(x);
        }
    }

    void pop() {
        if(data.top()==support.top())   support.pop();
        data.pop();
    }

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

    int getMin() {
        return support.top();
    }
};
int main()
{
    MinStack myMinStack;
    myMinStack.push(1);
    cout<<myMinStack.getMin()<<endl;
    myMinStack.push(2);
    cout<<myMinStack.getMin()<<endl;
    myMinStack.push(0);
    cout<<myMinStack.getMin()<<endl;
    myMinStack.pop();
    myMinStack.pop();
    cout<<myMinStack.getMin()<<endl;
    myMinStack.pop();
    return 0;
}