最小栈问题
题目:设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
个人思路:
需要俩个栈 一个普通栈 一个最小栈
push :普通栈一定放进去
如果最小栈为空 也放进去
如果最小栈不为空 和最小栈顶元素比较 如果大于栈顶元素,则不放
小于等于栈顶元素,也放进去
pop:普通栈顶元素小于等于最小栈栈顶元素时 最小栈栈顶元素也得删除
class MinStack {
private Stack<Integer> stack;
private Stack<Integer> minStack;
public MinStack() {
stack = new Stack<>();
minStack = new Stack<>();
}
public void push(int val) {
stack.push(val);
if(minStack.isEmpty()){
minStack.push(val);
}else{
if(val<=minStack.peek()){
minStack.push(val);
}
}
}
public void pop() {
int ret = stack.peek();
stack.pop();
if(ret<=minStack.peek()){
minStack.pop();
}
}
public int top() {
if(stack.isEmpty()){
return -1;
}
return stack.peek();
}
public int getMin() {
if(minStack.isEmpty()){
return -1;
}
return minStack.peek();
}
}