题目描述
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
实现 MinStack 类:
MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。示例 1:
输入:
[“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.提示:
-231 <= val <= 231 - 1
pop、top 和 getMin 操作总是在 非空栈 上调用
push, pop, top, and getMin最多被调用 3 * 104 次
做题思路
本题就是考察我们对栈和队列这两个经典的数据结构的掌握程度
栈:先进后出
最小栈可不等于上一个题目 “顺序栈”,可能有的人在做的时候,会维护一个顺序栈,让最小的值一直都在栈顶;
其实这个思路一开始就错了,题目中只是说在常数级别下找到栈中最小值,并没有说构造顺序栈,因为你构造一个顺序栈虽然可以在常数级找到最小值,但是你这个栈的"元素顺序"就完全乱了,因为你这个时候就不知道哪个元素先插入的,此时你做pop top 的时候就会出错
正确的思路应该是我们定义一个成员变量min,用min来记录最小值,还需要一个辅助栈帮助我们去找最小值
我们可以在每次插入的时候判断min和val的大小来确定最小值
当我们删除的时候就需要注意了,如果删除的val==min,那么此时我们就又需要去找min了。思路就是遍历stack1,在这个过程比较min和其元素大小,最后遍历完之后再将stack2里面元素放回stack1
代码实现
class MinStack {
Stack<Integer> stack1=new Stack<>();
Stack<Integer> stack2=new Stack<>();
int min=Integer.MAX_VALUE;
public MinStack() {
}
public void push(int val) {
stack1.push(val);
if(val<min)min=val;
}
public void pop() {
int x=stack1.pop();
if(x==min){
min=Integer.MAX_VALUE;
while(!stack1.isEmpty()){
int temp=stack1.peek();
if(temp<min)min=temp;
stack2.push(stack1.pop());
}
while(!stack2.isEmpty())stack1.push(stack2.pop());
}
}
public int top() {
return stack1.peek();
}
public int getMin() {
return min;
}
}
/**
* Your MinStack object will be instantiated and called as such:
* MinStack obj = new MinStack();
* obj.push(val);
* obj.pop();
* int param_3 = obj.top();
* int param_4 = obj.getMin();
*/