写一个算法编程题
面试题来源于力扣上的一道面试题
请设计一个栈,除了常规栈支持的pop与push函数以外,还支持min函数,该函数返回栈元素中的最小值。执行push、pop和min操作的时间复杂度必须为O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
JAVA 代码块
class MinStack {
private Stack<Integer> mainstack;
private Stack<Integer> minstack;
/** initialize your data structure here. */
public MinStack() {
mainstack=new Stack<Integer>();
minstack=new Stack<Integer>();
}
public void push(int x) {
if(mainstack.empty()){
mainstack.push(x);
minstack.push(x);
} else{
mainstack.push(x);
if(minstack.peek()<x){
minstack.push(minstack.peek());
}else {
minstack.push(x);
}
}
}
public void pop() {
mainstack.pop();
minstack.pop();
}
public int top() {
return mainstack.peek();
}
public int getMin() {
return minstack.peek();
}
}
思路就是利用辅助栈 来入出栈 进而达到效果 。
这是我的第一个博客写的不好多多指教