题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
定义一个辅助栈
每当push 一个元素时 如果当前辅助栈为空 或者当前元素小于辅助栈栈顶元素 往辅助栈也push这个元素。 否则 继续push辅助栈栈顶元素
每当pop一个元素时 辅助栈和原栈都需要pop
想要获取最小元素时只需要从辅助栈中 取元素
这样子做的话 无论何时 getMin都是最小元素
下面是java代码实现
public class Solution {
Stack<Integer> stack = new Stack();
Stack<Integer> temp = new Stack();
public void push(int node) {
stack.push(node);
if(temp.isEmpty() || temp.peek() >= node) {
temp.push(node);
}else {
temp.push(temp.peek());
}
}
public void pop() {
temp.pop();
stack.pop();
}
public int top() {
return stack.peek();
}
public int min() {
return temp.peek();
}
}