剑指offer 41. 包含min函数的栈-java实现

AcWing 41. 包含min函数的栈

原题链接

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

push(x)–将元素x插入栈中
pop()–移除栈顶元素
top()–得到栈顶元素
getMin()–得到栈中最小元素
数据范围
操作命令总数 [0,100]。

代码案例：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.

题解

辅助栈的做法 简单 就不在过于赘述 时间复杂度是O（1）
1 、在维护基础栈stack的同时还需要维护一个stackMin栈来记录stack栈的最小值
2、进栈push(int x)操作时，需要判断x与当前栈中最小值的大小，更新stack栈的最小值
3、出栈pop()操作时，需要stack栈的最小值
如果不通过辅助栈的方法 就需要把最小值记录一下就可以了 需要的时候返回Min

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