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

目录

一、题目

二、思路

入栈操作：

出栈操作：

三、代码

四、执行结果

---

一、题目

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

提示：

1. 各函数的调用总次数不超过 20000 次

二、思路

先定义两个栈，stack1、stack2，

入栈操作：

        栈1、栈2同时入栈，但是有一下区别：

栈1正常入栈，

栈2如果栈底为空就入栈，栈底不为空判断当前元素是否小于栈顶元素，小于或等于则入栈。

出栈操作：

   栈1、栈2同时出栈，但是有一下区别：

栈1正常出栈、

栈2是否出栈需判断当前元素是否等于栈2的栈顶元素。等于则出栈

三、代码

class MinStack {

    Stack<Integer> stack1;
     Stack<Integer> stack2;

    /** initialize your data structure here. */
    public MinStack() {
      this.stack1 =  new Stack();
      this.stack2 =  new Stack();

    }
    // 进栈操作
    public void push(int x) {
        stack1.push(x);
        if(stack2.isEmpty()||x<=stack2.peek()){
            stack2.push(x);
        }
    }

    // 出栈操作
    public void pop() {
        // 栈1为空则都为空
        if(!stack1.isEmpty()){
            // 数值>127时,自动装箱为Integer对象,此时比较的是内存地址,需要取到数值
            if(stack1.peek().intValue()==stack2.peek().intValue()){
                stack2.pop();
            }
            stack1.pop();
        }
    }
    // 获得最上面一个元素
    public int top() {
        return stack1.peek();
    }
    
    public int min() {
        return stack2.peek();
    }
}

四、执行结果