我的leetcode代码已放入github:https://github.com/gaohongbin/leetcode
题目:
Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.
- push(x) -- Push element x onto stack.
- pop() -- Removes the element on top of the stack.
- top() -- Get the top element.
- getMin() -- Retrieve the minimum element in the stack.
翻译:
设计一个栈,可以实现push,pop,top和getMin方法。
其中getMin()方法返回现在栈中的最小元素。
思想:
用两个栈stack和min,stack用来实现栈的正常功能,min栈用来装栈中的比较小的元素,当min栈为空时,如果push,则把元素放入min栈中,如果min栈不为空,则如果push的元素x小于等于min的栈顶元素,则将x压入min栈中,出栈时,如果min栈的最顶元素和stack栈的最顶元素相等,则两个都同时出栈,否则只有stack出栈。
getMin时,只需返回min栈顶元素即可。
代码:
class MinStack {
Stack<Integer> stack1=new Stack<Integer>();
Stack<Integer> min=new Stack<Integer>();
public void push(int x) {
stack1.push(x);
if(min.size()==0 || x<=min.peek())
min.push(x);
}
public void pop() {
if(min.size()>0 && stack1.peek().equals(min.peek()))
min.pop();
stack1.pop();
}
public int top() {
int x=stack1.peek();
return x;
}
public int getMin() {
return min.peek();
}
}