每日一刷：最小栈

乐心湖

发布于 2021-03-02 11:37:49

2720

发布于 2021-03-02 11:37:49

文章被收录于专栏：MyTechnology

自言自语

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

最小栈

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

push(x) —— 将元素 x 推入栈中。

pop() —— 删除栈顶的元素。

top() —— 获取栈顶元素。

getMin() —— 检索栈中的最小元素。

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/min-stack 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

比较容易，一个基本栈（正常操作），一个辅助栈（栈顶保存最小值）来获取最小元素。当 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();
  }
}