力扣 155. 最小栈

题目来源:https://leetcode-cn.com/problems/min-stack/

大致题意:
实现一个最小栈 MinStack 类:

  • MinStack() 初始化堆栈对象
  • void push(int val) 将元素val推入堆栈
  • void pop() 删除堆栈顶部的元素
  • int top() 获取堆栈顶部的元素
  • int getMin() 获取堆栈中的最小元素

思路

题目要求 getMin() 方法复杂度为常数级,所以需要在出入栈的时候维护最小值,可以使用一个辅助栈,栈顶元素为原数栈每个位置对应的栈中最小值

入栈:

  • 每次 val 入栈时,取辅助栈当前栈顶元素与 val 的最小值入辅助栈
  • 初始时,先给辅助栈放入一个整形的最大值做哨兵
    出栈:
  • 原始栈和辅助栈弹出栈顶元素
    获取最小值
  • 返回辅助栈栈顶元素
public class MinStack {

    Stack<Integer> stack;
    Stack<Integer> minStack;    // 辅助栈

    public MinStack() {
        stack = new Stack<>();
        minStack = new Stack<>();
        // 先放入一个整型最大值
        minStack.push(Integer.MAX_VALUE);
    }

    public void push(int val) {
        stack.push(val);
        // 将当前最小值放入辅助栈
        minStack.push(Math.min(minStack.peek(), val));
    }

    public void pop() {
        stack.pop();
        minStack.pop();
    }

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

    public int getMin() {
        return minStack.peek();
    }
}

你可能感兴趣的:(leetcode)