leetcode:Min Stack

最新推荐文章于 2021-05-13 13:11:37 发布

SCAU啊源

最新推荐文章于 2021-05-13 13:11:37 发布

阅读量435

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/chanrenyuan/article/details/41115879

leetcode 专栏收录该内容

168 篇文章

订阅专栏

本文介绍了一种特殊的数据结构——栈，该栈能在常数时间内完成进栈、出栈及获取当前栈内最小值的操作。通过维护两个栈：一个用于存储所有元素，另一个用于跟踪最小值，确保getMin方法的时间复杂度为O(1)。

题目要求定义一个数据结构，栈，具备进栈出栈和查找当前栈的最小值的方法

刚开始的时候getMin方法我是直接遍历栈中数据，很明显O(N)的查找超市了

那么getMin的方法必须是常量级别的，除了定义个栈之外，

我们另外定义一个最小值栈，其中在最小值栈的栈顶我们保存当前栈的最小值

很明显，我们需要维护最小值栈的栈顶一直是当前栈的最小值，两种操作

1进栈，当最小值栈为空，直接进栈，

当最小值栈的栈顶小于等于入栈的x值时候，最小值栈push(x)

2出栈当元素X出栈时候，判断X是否等于最小值栈的栈顶，是的话最小值栈栈顶元素出栈

大家可以想象一下进栈出栈的动作，试着理解算法的过程

不过做题的过程，我刚开始是用了LinkedList作为栈，但是OJ竟然判了我Memery 溢出

后来改用了ArrayList竟然Accept了，真心想不明白。。。。。。

class MinStack {
    
    ArrayList<Integer> valueStack = new ArrayList<>(100000);
    ArrayList<Integer> minValueStack = new ArrayList<>(100000);
    public void push(int x) {
        valueStack.add(x);
        if(minValueStack.isEmpty()) minValueStack.add(x);
        else if(x <= minValueStack.get(minValueStack.size() - 1)) minValueStack.add(x);
    }

    public void pop() {
        int value = valueStack.get(valueStack.size() - 1);
        valueStack.remove(valueStack.size() - 1);
        if(value == minValueStack.get(minValueStack.size() - 1)){
            minValueStack.remove(minValueStack.size() - 1);
        }
    }

    public int top() {
        return valueStack.get(valueStack.size() - 1);
    }

    public int getMin() {
        return minValueStack.get(minValueStack.size() - 1);
    }
}