题目描述:
设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
题解:
需要使用双栈实现,一个保存数据,一个保存最小值。由于随着数据出栈,最小值是不断变化的,因此需要一个最小值栈来保存最小值。
class MinStack {
Stack<Integer> stack=new Stack<Integer>();
Stack<Integer> minstack=new Stack<Integer>(); //建一个最小值栈保存最小值
/** initialize your data structure here. */
public MinStack() {
}
public void push(int x) {
//如果最小值栈为空或者新push的值比最小值栈的栈顶元素小则将此元素push进最小值栈
if(minstack.isEmpty() || x<=minstack.peek())
minstack.push(x);
stack.push(x); //普通栈push进值
}
public void pop() {
if(!stack.isEmpty()){
//如果普通栈的栈顶元素与最小值栈的栈顶元素相同,最小值栈pop出栈顶元素从而与普通栈保持一致
if(minstack.peek().equals(stack.peek()))
minstack.pop();
stack.pop();
}
}
public int top() {
return stack.peek();
}
public int getMin() {
return minstack.peek(); //最小值栈的栈顶元素就是最小值
}
}
/**
* Your MinStack object will be instantiated and called as such:
* MinStack obj = new MinStack();
* obj.push(x);
* obj.pop();
* int param_3 = obj.top();
* int param_4 = obj.getMin();
*/