[头条]笔试题——最小栈

 

 解法一：（相对解法二，会非常鸡肋）

思路：

• 由于获取堆栈最小元素时间复杂度为O(1)，所有用创建一个变量min用来存放最小值
• 将存放最小值的变量min设为Integer.MAX_VALUE
• push方法里，每push一次用min比较
• 创建两个栈，第一个栈用来存放数据，第二栈用来通过变量第一个栈而得出最小数据的零时栈

class MinStack {
    public Stack<Integer> stack;
    public Stack<Integer> stackTmp;
    public MinStack() {
        stack = new Stack<>();
        stackTmp = new Stack<>();
    }
    //先赋最大值
    public int min = Integer.MAX_VALUE;
    public void push(int val) {
        //有比他小的就赋值
        if(val < min){
            min = val;
        }
        stack.push(val);
    }
    
    public void pop() {
        if(stack.pop() == min){
            //被pop后min需要重新赋值，依旧用最大值
            min = Integer.MAX_VALUE;
            //将所有元素转移到stackTmp，目的是为了让min重新与所有元素比较
            while(!stack.empty()){
                int tmp = stack.pop();
                if(tmp < min){
                    min = tmp;
                }
                stackTmp.push(tmp);
            }
            //再转移到stack
            while(!stackTmp.empty()){
                stack.push(stackTmp.pop());
            }
        }
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int getMin() {
        return min;
    }
}

---

解法二：（分析如下）

 

 

 

 代码如下：

class MinStack {
    Stack<Integer> stack;//普通栈
    Stack<Integer> stackMin;//最小栈
    public MinStack() {
        stack = new Stack<>();
        stackMin = new Stack<>();
    }
    //压栈
    public void push(int val) {
        //给最小站赋初值
        if(stackMin.empty()){
            stackMin.push(val);
        }
        //若不是初值，先判断是否小于等于（等于是以防pop后为空）最小栈栈顶元素，若是，才给最小栈压栈
        else if(stackMin.peek() >= val){
                stackMin.push(val);
        }
        stack.push(val);
    }
    
    public void pop() {
        if((int)stackMin.peek() == (int)stack.peek()){
            stackMin.pop();
        }
        stack.pop();
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int getMin() {
        return stackMin.peek();
    }
}