题目:
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
解题思路:
定义两个栈,一个用来进行普通的栈操作(stack),另一个用来对stack栈中的最小值就行操作控制,具体实现见代码注释;
class MinStack {
private Stack<Integer> stack; //普通栈
private Stack<Integer> minStack; //最小值栈
public MinStack() {
stack=new Stack<>();
minStack=new Stack<>(); //构造方法:实例化两个栈
}
public void push(int val) { //压栈操作
stack.push(val); //val正常压入stack
if(!minStack.empty()){ //不是第一次的话
int top=minStack.peek(); //记录最小栈中的栈顶值
if(val<=top){ //如果目前压入的val小于或者等于top值,就把他也压入最小栈
minStack.push(val);
}
}else{ //第一次压栈时,minStack一定为空,直接压入即可
minStack.push(val);
}
}
public void pop() { //出栈操作
int top=stack.peek(); //记录stack栈顶值
int topMin=minStack.peek();//记录minStack栈顶值
stack.pop();//先进行普通栈的弹出操作
if(top==topMin){ //如果弹出的值与最小栈顶值也相同,也对最小栈就行弹出操作
minStack.pop();
}
}
public int top() { //正常取stack栈顶值即可
return stack.peek();
}
public int getMin() { //经过上面push与pop的操作。维护了minStack中栈顶值一定是最小值
return minStack.peek(); //弹出minStack栈顶值即可
}
}