剑指Offer JZ20 包含min函数的栈 C++实现

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O(1)）。

解题思路

---

方法一：辅助空间

1、思路：开辟一个辅助栈，辅助栈中存的是当前栈的最小值。做push操作时，当push的值小于辅助栈的最小值时，辅助栈将该值push到栈中，否则将栈顶再次push到栈中；做pop操作时，两个栈需要一起pop。

2、代码：

class Solution {
public:
    stack<int> valStack;
    stack<int> minStack;
    void push(int value) {
        valStack.push(value);
        if (minStack.empty() || value < minStack.top())
            minStack.push(value);
        else minStack.push(minStack.top());
    }
    void pop() {
        if (valStack.empty()) return;
        valStack.pop();
        minStack.pop();
    }
    int top() {
        if (valStack.empty()) return -1;
        return valStack.top();
    }
    int min() {
        if (minStack.empty()) return -1;
        return minStack.top();
    }
};

3、复杂度：

时间复杂度：O(1)；

空间复杂度：O(n)。