BM43-包含min函数的栈

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的 min 函数，输入操作时保证 pop、top 和 min 函数操作时，栈中一定有元素。

此栈包含的方法有：

• push(value):将value压入栈中
• pop():弹出栈顶元素
• top():获取栈顶元素
• min():获取栈中最小元素

数据范围：操作数量满足 0≤n≤300，输入的元素满足 ∣val∣≤10000。
进阶：栈的各个操作的时间复杂度是 O(1)，空间复杂度是 O(n)。

示例:

输入:    ["PSH-1","PSH2","MIN","TOP","POP","PSH1","TOP","MIN"]

输出:    -1,2,1,-1

解析:

• "PSH-1"表示将-1压入栈中，栈中元素为-1
• "PSH2"表示将2压入栈中，栈中元素为2，-1
• “MIN”表示获取此时栈中最小元素==>返回-1
• "TOP"表示获取栈顶元素==>返回2
• "POP"表示弹出栈顶元素，弹出2，栈中元素为-1
• "PSH1"表示将1压入栈中，栈中元素为1，-1
• "TOP"表示获取栈顶元素==>返回1
• “MIN”表示获取此时栈中最小元素==>返回-1

示例1

输入：["PSH-1","PSH2","MIN","TOP","POP","PSH1","TOP","MIN"]

返回值：-1,2,1,-1

---

思路：双栈实现

• element栈：具体存储元素。
• support栈：辅助栈，比较当前元素和此栈栈顶元素的值，取最小值放入此栈中，它就一直维护当前栈的最小值。

---

代码

import java.util.Stack;
 
public class Solution { 
    //具体存储元素
    private Stack<Integer> element = new Stack<>();
    //辅助栈，一直保存最小元素
    private Stack<Integer> support = new Stack<>();;
 
    public void push(int node) {
        element.push(node);
        if(support.isEmpty()) {
            support.push(node);
        } else {
            int tmpMin = support.peek();
            //比较当前元素和tmpMin谁更小，将最小元素push进辅助栈中
            int min = Math.min(node, tmpMin);
            support.push(min);
        }
    }
    
    public void pop() {
        element.pop();
        support.pop();
    }
    
    public int top() {
        return element.peek();
    }
    
    public int min() {
        return support.peek();
    }
}