[Lintcode]Min Stack

最新推荐文章于 2022-03-09 14:08:35 发布

原创最新推荐文章于 2022-03-09 14:08:35 发布 · 185 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode

算法专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种特殊栈的实现方法，该栈除了具备基本的push和pop操作外，还提供了一个min函数用于返回栈中当前的最小元素。所有操作的时间复杂度均为O(1)。通过使用两个栈来分别存储数据和维护最小值历史记录，从而实现了高效的数据管理和查询。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Implement a stack with min() function, which will return the smallest number in the stack.

It should support push, pop and min operation all in O(1) cost.

Example

push(1)
pop()   // return 1
push(2)
push(3)
min()   // return 2
push(1)
min()   // return 1

分析：两个stack。其中一个用来保存截止到每一个位置的最小值。不然，每次pop后，都需要重新搜索min

public class MinStack {
    
    Stack<Integer> res, min;

    public MinStack() {
        res = new Stack<Integer>();
        min = new Stack<Integer>();
    }

    public void push(int number) {
        res.push(new Integer(number));
        int minVal = min.isEmpty() ? number : Math.min(number, min.peek());
        min.push(new Integer(minVal));
    }

    public int pop() {
        min.pop();
        return res.pop();
    }

    public int min() {
        return min.peek();
    }
}