剑指Offer-包含min函数的栈

22.包含min函数的栈

设计一个支持push，pop，top等操作并且可以在O(1)时间内检索出最小元素的堆栈。

• push(x)–将元素x插入栈中
• pop()–移除栈顶元素
• top()–得到栈顶元素
• getMin()–得到栈中最小元素

样例

MinStack minStack = new MinStack();
minStack.push(-1);
minStack.push(3);
minStack.push(-4);
minStack.getMin();   --> Returns -4.
minStack.pop();
minStack.top();      --> Returns 3.
minStack.getMin();   --> Returns -1.

思路

这道题思路倒是不难，用两个栈就行了。一个栈stack1是普通的栈，保存压入的元素，另一个栈stack2保存当前stack1中所有元素的最小值。

push（）的具体做法是：直接压入stack1，如果stack2为空或者即将压入的元素小于stack2的栈顶元素，就直接压入stack2。反之，就压入stack2的栈顶元素（相当于复制了一个最小值）。这么做是为了在弹出元素的时候一样能保证stack2里的元素是最小的。因为stack1和stack2要一同弹出。

class MinStack {

    /** initialize your data structure here. */
    Stack<Integer> stack1=new Stack<Integer>();
    Stack<Integer>stack2=new Stack<Integer>();
    
    public MinStack() {
        
    }
    
    public void push(int x) {
        stack1.push(x);
        if(stack2.isEmpty()||x<=stack2.peek())
        {
            stack2.push(x);
        }else
        {
            stack2.push(stack2.peek());
        }
    }
    
    public void pop() {
        if(!stack1.isEmpty())
        {
            stack1.pop();
            stack2.pop(); 
        }      
    }
    
    public int top() {
        return stack1.peek();
    }
    
    public int getMin() {
        return stack2.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.getMin();
 */