Solution 2: Min Stack

问题描述

定义栈的数据结构，要求添加一个min函数，能够得到栈的最小元素。

实现栈的push(), pop()及getMin()函数，要求函数的时间复杂度为O(1).

解决思路

使用两个栈，一个为普通栈，实现push和pop函数；另一个记录所有入栈元素的非递增序列；

如下图所示：



程序

public class MinStack {
private Stack<Integer> normal;
private Stack<Integer> min;

public MinStack() {
normal = new Stack<Integer>();
min = new Stack<Integer>();
}

public void push(int elem) {
normal.push(elem);
if (min.isEmpty() || min.peek() >= elem) {
min.push(elem);
}
}

public int pop() throws Exception {
if (normal.isEmpty()) {
throw new Exception("Error: The Stack is Empty!");
}
int res = normal.pop();
if (min.peek() == res) {
min.pop();
}
return res;
}

public int getMin() throws Exception {
if (min.isEmpty()) {
throw new Exception("Error: The Stack is Empty!");
}
return min.peek();
}
}