题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
这个题要求时间复杂度是o(1)
竟然想了好久没想出来,原来思维定式了,一直在想既然要从小到大排序怎么可能实现o(1)呢
其实由于出栈入栈是由顺序的,用来记录最小值的栈也不用完整记录所有元素的顺序,只要根据入栈的顺序来就行
比如 入栈 9 8 10 4 5 2
如果入栈是发现比辅助栈的头元素小就入辅助栈
辅助栈 9 8 4 2
完全不懂担心 5没有加入栈中 因为只要前面有比它小的数,那它就永远不可能成为最小值
Stack<Integer> minStack = new Stack<>();
Stack<Integer> stack = new Stack<>();
public void push(int node) {
stack.push(node);
if(minStack.isEmpty() || node < minStack.peek()) minStack.push(node);
}
public void pop() {
if(minStack.peek() == stack.pop())
minStack.pop();
}
public int top() {
return stack.peek();
}
public int min() {
return minStack.peek();
}