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

题目

包含min函数的栈

思路

使用两个栈A,B，栈A正常完成push,pop,top操作；

栈B用于维护非严格降序的栈，即在push操作时，判断此次push的元素是否小于等于当前栈B的栈顶元素，若是，则压入栈B，否则不压入。

代码

class MinStack {

    Stack<Integer> A,B;
    /** initialize your data structure here. */
    public MinStack() {
        A=new Stack<>();
        B=new Stack<>();
    }
    
    //当要压入的元素小于等于栈B的栈顶元素，则压入
    //同时需要注意，栈B为空时就不需要进行判断了，直接压入即可
    public void push(int x) {
        A.push(x);
        
        if(B.isEmpty() || x<=B.peek())
        {
            B.push(x);
        }
    }
    
    //由于栈中为Integer对象，因此需要使用equals(),==的范围仅能保证-127~128相等
    public void pop() {

        if(A.peek().equals(B.peek()))
        {
            B.pop();
        }
        A.pop();
    }
    
    public int top() {
        return A.peek();
    }
    
    public int min() {
        return B.peek();
    }
}

/**
 * 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();
 */