一、知识准备
(1)栈:
- 核心:“先进后出”,从栈顶入,栈顶出。
-
栈的几种基本操作:void push(int data)--入栈,int pop()--弹出栈顶元素(即 删除并返回最后一个入栈的元素),int peek() --返回栈顶元素但不删除,boolean empty()--判断是否为空
(2)什么是最小栈?
二、思路:辅助栈minStack
因为找栈内最小元素每次都要遍历整个栈,时间复杂度过大,所以除了有一个普通栈你stack以外,多准备一个辅助栈minStack。辅助栈minStack--用来保存当前栈内最小元素。
①入栈:
- 普通栈stack里面所有元素都放,但是最小栈minStack里面只能放当前栈中最小的元素。
- 每次都放入stack后,每次都要看一下是否比当前栈里面最小的元素(即minStack栈顶元素)小,如果小于当前栈中最小的元素,那么则放入最小栈minStack,若大于则不放入minStack。
②出栈:
- 每次普通栈stack出元素后,都要判断 最小栈minStack栈顶元素 和 普通栈stack里面要出的元素是否一致,如果一致则minStack最小栈也出栈,如果不是则只出stack不出minStack。eg:stack出6的时候 minStack栈顶元素是2, 不是6,所以不用出。
- 假如没有判断,导致minStack没出而普通栈stack出了,那么 此时普通栈里面的最小元素 就和minStack栈顶元素不同,所以就不是最小栈了。
③图解:
三、代码实现
设计一个支持 push
,pop
,top
操作,并能在常数时间内检索到最小元素的栈。
public class MinStack {
private Stack<Integer> stack;//普通栈
private Stack<Integer> minStack;//辅助栈(当前栈内最小元素)
public MinStack() {
this.stack = new Stack<>();//初始化
this.minStack = new Stack<>();
}
public void push(int val) {
stack.push(val);
if(minStack.empty()) {
minStack.push(val);
}else {
int x = minStack.peek();
if (x >= val) {//这个等号必须取!!!!
minStack.push(val);
}else {
return;
}
}
}
public void pop() {
if(stack.empty()) {
return;
}
int x = stack.pop();
//普通栈不为空说明此时最小栈 一定不为空
if(x == minStack.peek()) {
minStack.pop();
}else {
return;
}
}
/**
* 只是获取普通栈的栈顶元素
* @return
*/
public int top() {
if(stack.empty()) {//普通栈为空
return -1;
}
return stack.peek();
}
/**
* 获取当前栈的最小元素
* @return
*/
public int getMin() {
if(minStack.empty()) {//辅助栈为空
return -1;
}
return minStack.peek();
}
}
四、注意
- 出栈、入栈里面 的 “每次都要比较判断”
-
“push方法中等号必须取!”----如果入栈的数据 和 minStack栈顶元素大小相等,则必须也要入栈,否则会影响 出栈时 的下一次栈内最小元素的比较。(见三中代码push方法 注释)