题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
题目分析
看到题目,第一反应就是使用一个整数来保存栈中的最小元素,这样可以时间入栈(push)函数、和min函数的时间复杂度为O(1),但却不能保证出栈(pop)函数的时间复杂度,因为当最小元素出栈后,我们不得不遍历栈来寻找次小值元素(即当前新的最小值元素),因此这不是我们想要的解法。
我们不妨换一种思路,既然最小元素出栈后,我们需要重新获取次小值元素,而这步操作导致我们时间复杂度不符合要求,那我们何不把次小值元素也保存起来?
首先,我们需要一个数据栈和辅助栈。先往空的数据栈里压入数字3,此时最小值是3,因此将3压入辅助栈。接下来向数据栈压入4,此时最小值为3,因此向辅助栈压入3。第三步继续向数据栈压入2,此时最小值为2,因此向辅助栈压入2。
由于我们每次向辅助栈压入的都是压入时的最小值,因此min函数中只需要返回辅助栈的栈顶元素即可。对于pop函数,每次对数据栈的pop操作,我们需要同时对辅助栈也进行以此pop操作,当数据栈pop出的元素是最小值时,辅助栈也对应将栈顶最小值pop出栈,剩下的即为次小值;当数据栈pop出的元素不是最小值时,辅助栈在pop后,栈顶元素也仍为最小值。因此这是一个正确的解法。
代码如下。
Java代码
import java.util.Stack;
public class Solution {
private Stack<Integer> stack = new Stack<Integer>();
private Stack<Integer> minStack = new Stack<Integer>();
public void push(int node) {
stack.push(node);
if(minStack.empty()){
minStack.push(node);
}else{
int min = minStack.peek()>node?node:minStack.peek();
minStack.push(min);
}
}
public void pop() {
stack.pop();
minStack.pop();
}
public int top() {
return stack.peek();
}
public int min() {
return minStack.peek();
}
}