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();}
}