29.包含min函数的栈

设计一个支持push,pop,top等操作并且可以在O(1)时间内检索出最小元素的堆栈。

  • push(x)–将元素x插入栈中
  • pop()–移除栈顶元素
  • top()–得到栈顶元素
  • getMin()–得到栈中最小元素

数据范围:

操作命令总数 [0,100]。

样例:

MinStack minStack = new MinStack();
minStack.push(-1);
minStack.push(3);
minStack.push(-4);
minStack.getMin(); --> Returns -4.
minStack.pop();
minStack.top(); --> Returns 3.
minStack.getMin(); --> Returns -1.

代码:

class MinStack {
    //主栈,存储所有元素
    private Deque<Integer>stack;
    //辅助栈,存储当前最小值
    private Deque<Integer>minStack;
    /** initialize your data structure here. */
    public MinStack() {
        //初始化主栈和辅助栈
        stack = new LinkedList<>();
        minStack = new LinkedList<>();
        //辅助栈初始时存储整型最大值
        minStack.push(Integer.MAX_VALUE);
    }
    
    public void push(int x) {
        //元素压入主栈
        stack.push(x);
        //新值小于辅助栈当前最小值时,压入新值
        //否则压入辅助栈当前最小值
        minStack.push(Math.min(minStack.peek(),x));
    }
    
    public void pop() {
        //弹出主栈元素
        stack.pop();
        //弹出辅助栈元素
        minStack.pop();
    }
    
    public int top() {
        //返回主栈栈顶元素
        return stack.peek();
    }
    
    public int getMin() {
        //返回辅助栈栈顶元素(当前最小值)
        return minStack.peek();
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */
posted @ 2025-05-22 09:25  回忆、少年  阅读(8)  评论(0)    收藏  举报