20、定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
思路:实现获取栈中的最小元素的min函数,若只用一个变量来保存当前栈中的最小元素,若该最小元素弹出了则无法知道栈中当前的最小元素是什么了,所以我们借助一个辅助栈来保存当前栈中的最小元素,压栈时若压栈元素小于最小栈最上面的元素则同时压入最小栈,弹出时,若弹出元素等于最小栈最上面的元素则同时把该元素弹出。要利用栈后进先出的特点,画图看会更清晰,更容易理解。
import java.util.Stack;
public class Solution {
//借助辅助栈来解决问题
Stack<Integer> stack1 = new Stack<>();
Stack<Integer> minStack = new Stack<>(); //用来存放最小栈元素
public void push(int node) {
//存入一个节点
stack1.push(node);
if(minStack.empty()){
minStack.push(node);
}else{
if(node<minStack.peek()){
minStack.push(node);
}
}
}
public void pop() {
//弹出一个元素
if(stack1.peek()==minStack.peek()){
stack1.pop();
minStack.pop();
}else{
stack1.pop();
}
}
public int top() {
return stack1.peek();
}
public int min() {
//获得栈中的最小元素
return minStack.peek();
}
}
peek()方法只获取stack的顶端元素,不将其从栈中删除。