20.包含min函数的栈
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
算法描述:
1.定义两个栈,一个是存传入的数据栈dataStack,一个是存储最小元素的栈minStack。
2.push操作:每次传入的数据node都push进dataStack栈中。然后判断minStack栈是否为空,如果为空则把node数据push进minStack栈中;如果不为空,就将传入的数据node和minStack中栈顶的元素比较,如果node较小或等于,则push进minStack,否则就不压进栈minStack。
3.pop操作:当数据栈要弹出元素时!!!为了保证辅助栈的栈顶元素始终是最小值,需要对辅助栈同时进行一个弹栈操作。但是首先判断两个栈的栈顶元素是否相等,相等话则辅助栈才执行弹栈操作,否则最小栈不用执行弹栈操作。
下图展示了栈内压入3、4、2、1之后接连两次弹出栈顶数字再压入0时,数据栈、辅助栈和最小值的状态。
例如:data中依次入栈,5, 4, 3, 8, 10, 11, 12, 1
则min依次入栈,5, 4, 3,no,no, no, no, 1,依次找次小值,只能找到有限个,并且与入栈元素顺序有关。
import java.util.Stack;
public class Solution {
Stack<Integer> dataStack =new Stack<Integer>();
Stack<Integer> minStack =new Stack<Integer>();
public void push(int node) {
dataStack.push(node);
if(minStack.isEmpty()){
minStack.push(node);
}
else if(node<=minStack.peek()){
//peek函数是取minstack的栈顶元素。
minStack.push(node);
}
}
public void pop(){
if(dataStack.peek() == minStack.peek())
minStack.pop();
dataStack.pop();
}
public int min(){
return minStack.peek();
}
}