题目
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
思路
这里主要是需要比原本的栈多出一个功能,就是常数时间内能检索栈中的最小元素。
主要是如何能记住栈中最小元素,这个问题就迎刃而解了
这里可以有三种方法
- 使用额外的栈,用来存放最小元素
- 在同一个栈中,标记最小元素,因为当抛出元素是最小元素时,需要记录之后的最小元素。判断push进来的元素是不是最小,如果是最小的话,push俩个元素,一个是之前的最小元素,另一个就是push进来的元素。这样当pop元素是最小的话,就记住了之前的最小元素。
- 使用链表来表示栈,结点里面还有一个min的变量用来表示最小元素
代码2
class MinStack {
class Node{
int value;
int min;
Node next;
Node(int x, int min){
this.value=x;
this.min=min;
next = null;
}
}
Node head = null;
//每次加入的节点放到头部
public void push(int x) {
if (null == head){
head = new Node(x,x);
} else {
//当前值和之前头结点的最小值较小的做为当前的 min
Node n = new Node(x, Math.min(x,head.min));
n.next=head;
head=n;
}
}
public void pop() {
if(head!=null)
head =head.next;
}
public int top() {
if(head!=null)
return head.value;
return -1;
}
public int getMin() {
if(null != head)
return head.min;
return -1;
}
}
代码三
class MinStack {
int min = Integer.MAX_VALUE;
Stack<Integer> MinStack = new Stack<>();
public void push(int x) {
if (x <= min) { //等于时也算进来一个最小元素
MinStack.push(min);
min=x;
}
MinStack.push(x);
}
public void pop() {
if(MinStack.pop() == min) {
min = MinStack.pop();
}
}
public int top() {
return MinStack.peek();
}
public int getMin() {
return min;
}
}