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

1.题目地址

力扣

2.题目描述

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

注意：本题与主站 155 题相同：https://leetcode-cn.com/problems/min-stack/

3. 思路

定义两个栈Stack1作为主栈，负责数据插入，弹出。Stack2作为辅助栈，存放主栈中非严格降序的元素。当一个数据进来时，先插入栈Stack1，再与栈Stack2的栈顶元素判断大小，若小于栈顶元素则入栈，因此保证Stack2中的栈顶元素始终是主栈Stack1中的最小元素。

力扣图示

 

4.代码

public class Offer30 {
    //主栈
    LinkedList<Integer> Stack1;
    //辅助
    LinkedList<Integer> Stack2;

    public Offer30() {
        Stack1 = new LinkedList<>();
        Stack2 = new LinkedList<>();
    }

    public void push(int x) {
        Stack1.push(x);
        //peek()查看栈顶元素而不移除它,与栈顶元素比较最小值入栈
        if (Stack2.isEmpty() || Stack2.peek() >= x) {
            Stack2.push(x);
        }
    }

    public void pop() {
        if (Stack1.pop().equals(Stack2.peek())) {
            Stack2.pop();
        }
    }

    public int top() {
        return Stack1.peek();
    }

    public int min() {
        return Stack2.peek();
    }
}