C++算法（六）包含 min 函数的栈

---

包含 min 函数的栈

一、题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.

作者：Krahets
链接：https://leetcode-cn.com/leetbook/read/illustration-of-algorithm/50bp33/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

二、解题思路及代码实现

1、解题思路

不打算使用c++的stack库容器，所以自己设计了一个双数据域的单向列表，一个数据存值，一个数据存当前栈的最小值。优点是执行速度还是挺快的，缺点是占用内存多。如果采用辅助列表的方式，和官方给的题解一样实现，确实能省一点内存。以下代码为我设计的栈，非最优解。

2、C++代码实现

class MinStack {
private:
    struct ListNode {
          int val;
          int min;
          ListNode *next;
          ListNode(int x, int y) : val(x), min(y), next(NULL) {}
    };

    ListNode* m_head;
    int m_min;


public:
    /** initialize your data structure here. */
    MinStack() {
        m_head = NULL;
    }

    ~MinStack() {
        while (m_head != NULL) {
            ListNode* tmp = m_head;
            m_head = m_head->next;
            delete tmp;
        }
    }

    void push(int x) {
        if (m_head == NULL) {
            m_head = new ListNode(x, x);
        } else {
            int tmp = x > m_head->min ? m_head->min : x;
            ListNode* new_node = new ListNode(x, tmp);
            new_node->next = m_head;
            m_head = new_node;
        }
    }

    void pop() {
        if (m_head == NULL) {
            return;
        } else {
            ListNode* tmp = m_head;
            m_head = m_head->next;
            delete tmp;
        }
    }

    int top() {
        if (m_head == NULL) {
            throw out_of_range("栈已空，操作无效！");
        }

        return m_head->val;
    }

    int min() {
        if (m_head == NULL) {
            throw out_of_range("栈已空，操作无效！");
        }

        return m_head->min;
    }
};

三、提交结果

---

总结

栈是一种后进先出的数据结构，c++类库并没有实现带min函数的栈，不过，按官方的解题思路，使用一个辅助栈来存最小值，确实能省一点内存，最坏的情况是每个元素都存，就和我设计的列表内存占用差不多了。