【重温经典】最小栈
方法1:辅助栈[浪费空间]
-
基础版辅助栈,准备一个
data
, 数据栈,准备一个help
辅助栈 -
其中
data
的存数据的,help
辅助栈用来存最小值,在push
操作时,help
如果栈顶元素大于待push
的元素,将待push
的元素塞进help
中,如果不是,则重复塞一次help
的栈顶元素,注意help
为空的时候特殊处理下
- 准备两个栈,data和help,做push操作时,需要保持help栈顶的元素始终最小,data的数据正常推入,help栈顶维持最小,在执行getMin方法的时候,返回help的栈顶元素
class MinStack {
private Stack<Integer> data;
private Stack<Integer> help;
/**
* initialize your data structure here.
*/
public MinStack() {
data = new Stack<>();
help = new Stack<>();
}
public void push(int x) {
data.push(x);
if (help.isEmpty()) help.push(x);
else if (help.peek() < x) help.push(help.peek());
else help.push(x);
}
public void pop() {
if (data.isEmpty()) throw new RuntimeException("stack empty");
data.pop();
help.pop();
}
public int top() {
if (data.isEmpty()) throw new RuntimeException("stack empty");
return data.peek();
}
public int getMin() {
if (data.isEmpty()) throw new RuntimeException("stack empty");
return help.peek();
}
}
方法2:辅助栈[不浪费空间]
- 升级版本辅助栈,当
push
进去的时候,- 当
help
的栈顶元素比新来的元素小的时候,这个时候保持help
不变 - 当
help
的栈顶元素大于等于新来的元素时,help
同步要推一份新的元素进来 - 当
help
元素为空的时候,也需要往里推
- 当
pop
的时候,pop
的元素是否和help
的元素有重叠,有就将help
的元素pop
出去,没有就维持不动,data
栈正常pop
class MinStack {
Stack<Integer> data;
Stack<Integer> help;
/**
* initialize your data structure here.
*/
public MinStack() {
data = new Stack<>();
help = new Stack<>();
}
public void push(int x) {
if (help.isEmpty() || help.peek() >= x) {
help.push(x);
}
data.push(x);
}
public void pop() {
if (data.isEmpty()) throw new RuntimeException("The stack is empty");
int pop = data.pop();
if (pop == help.peek()) {
help.pop();
}
}
public int top() {
if (data.isEmpty()) throw new RuntimeException("The stack is empty");
return data.peek();
}
public int getMin() {
if (help.isEmpty()) throw new RuntimeException("The stack is empty");
return help.peek();
}
}