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

剑指 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.

提示：

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

解题思路

使用两个栈来实现最小元素min（）函数

• stack1 作为主栈，进行push（）、pop () 和 top（）函数
• stack2 存储stack1中所有非严格降序元素的子序列 。 stack1中的最小元素始终对应stack2中的栈顶元素

函数设计

1. push(x)函数 ：

   • stack1直接push（x）
   • 当stack2大小为0时，说明当前并没有最小值，直接stack2.push(x)
   • 当x < stack2.peek() 当前最小值的时候，stack2.push(x)，更新最小值

2. top（）函数 :

   return stack1.peek()

3. min() 函数 ：

   return stack2.peek();

4. pop（）函数 ：

   当pop（）的元素是当前最小元素的时候，必须保证stack2.pop()，stack2及时更新

代码实现

class MinStack{
    private Stack<Integer> minstack; //stack2 最小值栈
    private Stack<Integer> stack;  //stack1 主栈
    public MinStack(){
        this.stack = new LinkedList<Integer>();
        this.minstack = new LinkedList<Integer>();
    }
    public void push(int x)
    {
        this.stack.push(x);
 		if(this.minstack.size()==0)
        {
            this.minstack.push(x);
        }else if(x<=this.minstack.peek()) //重复值也参与计算
        {
            this.minstack.push(x); 
        }
    }
    public void pop(){
        if(this.stack.size()>0)
        {
            int remove = this.stack.pop();
            if(remove==this.minstack.pop())
            {
                this.minstack.pop();
            }
        }
    }
    public int top(){
        return this.stack.peek();
    }
    public int min(){
        return this.minstack.peek();
    }
}