思路如下:
1、需要检索栈中最小元素,因此需要一个辅助栈保存最小值。
2、Stack中的peek()方法与pop()方法的区别:两者均会返回栈顶元素,pop()方法会将栈顶元素删除,而peek()不会。
3、pop()方法中,如果删除了stack中的栈顶元素是最小值的话,辅助栈的最小值也要删除。
class MinStack {
Stack<Integer> stack=new Stack<>();
//辅助栈存储最小值
Stack<Integer> min=new Stack<>();
/** initialize your data structure here. */
public MinStack() {
}
public void push(int x) {
stack.push(x);
//如果x是最小值,那么把x存入min辅助栈,>=是为了保证相同的最小值也能够存入栈中
if(min.isEmpty()||min.peek()>=x){
min.push(x);
}
}
public void pop() {
int x=stack.pop();
//如果删除的是最小值,那么辅助栈的最小值(栈顶元素)也要删除
if(x==min.peek()){
min.pop();
}
}
public int top() {
return stack.peek();
}
public int getMin() {
//获取最小值
return min.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();
*/
原题地址:
155. 最小栈