初级算法之设计问题:最小栈

设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。

push(x) -- 将元素 x 推入栈中。
pop() -- 删除栈顶的元素。
top() -- 获取栈顶元素。
getMin() -- 检索栈中的最小元素。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

思路:使用一个记录最小元素的栈+一个记录所有数的栈,每次有比最小栈小的数就压入,pop时,检验一下最小栈是否也需要pop

public class MinStack {
	public ArrayDeque stack = new ArrayDeque<>();
	public ArrayDeque minStack = new ArrayDeque<>();// 只存储由大到小的栈,比栈顶大的就不存了

	public MinStack() {
	}

	public void push(int x) {
		stack.push(x);
		if (minStack.peek() == null || (int) minStack.peek() >= x) {
			minStack.push(x);
		}
	}

	public void pop() {
		int temp = (int) stack.pop();
		if ((int) minStack.peek() == temp)
			minStack.pop();
	}

	public int top() {
		return (int) stack.peek();
	}

	public int getMin() {
		if (minStack.size() != 0)
			return (int) minStack.peek();
		return 0;
	}

}

注意点:
(1)对于栈的使用,通过双向队列ArrayDeque实现
(2)if (minStack.peek() == null || (int) minStack.peek() >= x) 第一次写的时候没有写>=而是>,导致pop的时候出错

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值