题目: 实现一个栈,该栈带有出栈(pop)、入栈(push)、取最小元素(getMin)3个方法,要保证这3个方法的时间复杂度都是O(1)
解题思路:
- 设原有的栈叫做栈A,此时创建一个额外的“备胎”栈B,用于辅助栈A;
- 当第一个元素进入栈A,让新元素也进入栈B。这个唯一的元素就是栈A的当前最小值。
- 之后每当新元素进入栈A时,比较新元素和栈A当前最小值的大小,如果小于栈A当前最小值,则让新元素进入栈B,此时栈B的栈顶元素就是栈A当前最小值;
- 每当栈A有元素出栈时,如果出栈元素是栈A当前最小值,则让栈B的栈顶元素也出栈。此时栈B余下的栈顶元素所指向的是栈A当中原本第2小的元素,代替刚才出栈的元素成为最小值。
代码实现:
private Stack<Integer> mainStack=new Stack<>();//主栈
private Stack<Integer> minStack=new Stack<>();//辅助栈
/**
* 入栈操作
* @param element 入栈的元素
*/
public void push(int element){
mainStack.push(element);
//如果辅助栈为空,或者新元素小于或等于辅助栈栈顶,则将新元素压入辅助栈
if (minStack.empty()||element<=minStack.peek()){
minStack.push(element);
}
}
//出栈操作
public Integer pop(){
//如果出栈元素和辅助栈顶元素值相等,辅助栈出栈
if (mainStack.peek().equals(minStack.peek())){
minStack.pop();
}
return mainStack.pop();
}
//获取栈中的最小元素
public int getMin() throws Exception {
if (mainStack.empty()){
throw new Exception("stack is empty");
}
return minStack.peek();
}