题目描述:
实现一个带有取最小值min方法的栈,min方法将返回当前栈中的最小值。
你实现的栈将支持push,pop 和 min 操作,所有操作要求都在O(1)时间内完成。
思路:
- 添加辅助栈
- 往工作栈里压入数据,若数据小于原有栈中的数,则往辅助栈中压入该数据,否则往辅助栈里压入一个之前判断为最小的数据
- 弹出数据时,工作栈和辅助栈同时弹出一个数
获取最小值就是获取辅助栈栈顶元素
Stack<Integer> stack1 = new Stack<>();
Stack<Integer> stack2 = new Stack<>();
int min;
public void push(int node) {
if(stack1.isEmpty()){
stack1.push(node);
stack2.push(node);
min = node;
}
else{
min = min()//取到目前的最小值,因为保存在min变量中的值可能已经被弹出了
if(node<min){
min = node;
stack1.push(node);
stack2.push(node);
}
else{
stack1.push(node);
stack2.push(min);
}
}
}
public void pop() {
stack1.pop();
stack2.pop();
}
public int top() {
return stack1.peek();
}
public int min() {
return stack2.peek();
}
}