每日一题剑指2

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 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.

两个栈，比较简单

class MinStack {

public:

    /** initialize your data structure here. */

            stack<int> stack1;

            stack<int> stack2;

    MinStack() {

    }

    

    void push(int x) {

        stack1.push(x);

        if(stack2.empty()){

            stack2.push(x);

        }else if (stack2.top()>=x){

            stack2.push(x);

        }

    }

    

    void pop() {

        if (stack2.top()==stack1.top()){

            stack2.pop();

        }

        stack1.pop();

    }

    

    int top() {

        return stack1.top();

    }

    

    int min() {

        return stack2.top();

    }

};