LeetCode刷题之路-栈-155. 最小栈

前言

感谢队友推荐的学习群。今天的佛系刷题是：155 最小栈。

题目描述

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

push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。

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

解题思路

最开始我是想用数组实现的，但是在写的时候发现很麻烦：入栈要考虑扩容问题，出栈要考虑数组元素迁移问题，还要保证值最小的同步变化。写不下去之后，决定用链表。
开始时，用单向链表，但是发现出栈变得麻烦了。于是改用双向链表。至于最小值，我用了一种很取巧的办法：在每个Node中都定义一个最小值。每次都之和前一个元素比较最小值，并且只更新当前插入元素的最小值。这样，即使出栈，也不需要重新处理。直接就是最后一个元素的最小值。

代码实现

class Node {
        int val;
        int min;
        Node next;
        Node pre;

        Node(int val, int min) {
            this.val = val;
            this.min = min;
        }
    }

    private Node tail;

    /**
     * initialize your data structure here.
     */
    public MinStack() {
    }

    public void push(int val) {
        Node ele = new Node(val, val);
        if (tail != null) {
            if (tail.min < val) {
                // 只需更新最后的元素的最小值，因为会出栈，避免排序
                ele.min = tail.min;
            }
            tail.next = ele;
            ele.pre = tail;
            tail = tail.next;
        }
        tail = ele;
    }


    public void pop() {
        tail = tail.pre;
    }

    public int top() {
        return tail.val;
    }

    public int getMin() {
        return tail.min;
    }

小结

官方的实现实际上是直接使用了java中的双端队列Deque - LinkedList。但是它需要用2个双端队列。而我的实现，是通过自定义Node，把最小值放到了每个元素里面了。本质上都是差不多的。每个元素都对应一个最小值。