题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
思路分析
使用自带的stack函数和属性。
- 建立两个Stack,一个存放数据的栈,一个存放最小值的栈
- push时,数据栈直接push,最小值栈需要判断是否空,空则直接push,非空判断minStack的栈顶元素值和node值的大小。栈顶大就再压入一次栈顶元素,node大就压入node。
- pop两个栈直接弹出
- top为取数据栈的栈顶元素
- min为取最小值的栈顶元素
代码实现
Stack<Integer> dataStack = new Stack<>();
Stack<Integer> minStack = new Stack<>();
public void push(int node) {
dataStack.push(node);
if (minStack.empty()) {
minStack.push(node);
} else if (minStack.peek() >= node) {
minStack.push(node);
}else {
minStack.push(minStack.peek());
}
}
public void pop() {
dataStack.pop();
minStack.pop();
}
public int top() {
return dataStack.peek();
}
public int min() {
return minStack.peek();
}