剑指 Offer每日打卡

第 1 天栈与队列（简单）

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

解题思路

普通栈的 push() 和 pop() 函数的复杂度为 O(1)O(1) ；

而获取栈最小值 min() 函数需要遍历整个栈，复杂度为 O(N)O(N) 。

本题难点： 将 min() 函数复杂度降为 O(1)O(1) ，可通过建立辅助栈实现

题解：

class MinStack {
    /** initialize your data structure here. */
    Stack<Integer> A,B;
    public MinStack() {
        A = new Stack<>();
        B = new Stack<>();
    }
    
    public void push(int x) {
        A.push(x);
        if(B.empty()||B.peek()>=x){
            B.push(x);
        }
    }
    
    public void pop() {
        //注意当AB栈都存在x时，B才进行删除
       if(A.pop().equals(B.peek())){
           B.pop();
       }
    }
    
    public int top() {
        return A.peek();
    }
    
    public int min() {
        return B.peek();
    }
}

Stack知识补充

• 先入后出，添加和删除的复杂度都为O(1)，查询为O(n)

• 基本操作：入栈，出栈都是针对栈顶元素，因为栈是无序的，既需要遍历所以查询某个节点为O(n)

• 结构：队列数组
• 特点：线程安全，是Vector的子类，有自己特有的方法

boolean	empty()           测试堆栈是否为空。
E	peek()           查看堆栈顶部的对象，但不从堆栈中移除它。
E	pop()           移除堆栈顶部的对象，并作为此函数的值返回该对象。
E	push(E item)           把项压入堆栈顶部。
int	search(Object o)           返回对象在堆栈中的位置，以 1 为基数。