不管遇到什么挫折,明天的太阳都会照样升起。

最小栈

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

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

比较容易,一个基本栈(正常操作),一个辅助栈(栈顶保存最小值)来获取最小元素。当 push 的元素小于等于辅助栈栈顶时,存入辅助站即可,这样辅助站栈顶永远是我们的最小元素。pop 需要考虑当原栈顶跟辅助栈顶相同的情况。

class MinStack {

  public Stack<Integer> stack;
  public Stack<Integer> minStack;

  /**
     * initialize your data structure here.
     */
  public MinStack() {
    stack = new Stack<>();
    minStack = new Stack<>();
  }

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

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

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

  public int getMin() {
    return minStack.peek();
  }
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-stack
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

Last modification:April 15, 2021
如果觉得我的文章对你有用,请随意赞赏