设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
示例:
输入:
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]
输出:
[null,null,null,null,-3,null,0,-2]
解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
思路:这道题不是让你去自己实现一个栈,所以可以直接用java类库中的stack或者LinkedList类直接调用push和pop。
关键是getmin()方法,我们另外建立一个minstack栈,其中的元素与stack栈内元素一一对应,保存stack中对应元素为栈顶时的栈内最小值,stack入栈一个元素,它也入栈一个元素,stack出栈一个元素,它也出栈一个元素。
stack 入栈元素T, 比较minstack栈顶元素和T 的大小,将小的那个压入minstack。
stack 出栈,将minstack也出栈。
getmin(): return minstack.peek();
注意:这是优先队列的方法
class MinStack {
LinkedList<Integer> stack;
LinkedList<Integer> minStack;//minstack 保存每一个stack中的元素所对应的当前最小值(当它为栈顶的时候的栈内最小值)
//每次stack入栈时,minstack也将当前栈内最小值入栈
//每次stack出栈时,minstack也将当前栈内最小值出栈
/** initialize your data structure here. */
public MinStack() {
stack = new LinkedList<>();
minStack = new LinkedList<>();
}
public void push(int val) {
if(stack.isEmpty())
{
stack.push(val);
minStack.push(val);
return;
}
stack.push(val);
int min = minStack.peek();//当前栈内最小值
int minpush = (val<min)?val:min;
//栈不空的时候需要和当前最小值进行比较,如果新入栈的元素小于当前最小值,则它对应的栈内最小值为它的值
minStack.push(minpush);//将新的栈内最小值入栈
}
public void pop() {
if(!stack.isEmpty())
{
stack.pop();
minStack.pop();
}
}
public int top() {
return stack.peek();
}
public int getMin() {
return minStack.peek();
}
}