剑指Offer算法实现之二十一：包含min函数的栈

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

思路：

min值是变化的，并且当前min弹出后新的min是是什么？
直接的想法：用一个min栈专门跟踪每个push操作后对应的最小值。因此min栈与数据栈占用空间相同
能否优化空间？仅当push元素x<=min.top()时才将x入栈。每次pop时，判断min.top()是否与data().top()是否相同，若是，则min出栈。

编译环境：ArchLinux+Clang3.3，C++11

实现一：

#include <iostream>
#include <stack>
#include <stdexcept>
#include <cassert>
using namespace std;

template<typename T>
class StackWithMin {
public:
    void push(const T value);
    void pop();
    const T min() const;
private:
    stack<T> m_data; // 数据栈
    stack<T> m_min; // min跟踪栈
};
template<typename T>
void StackWithMin<T>::push(const T value)
{
    m_data.push(value);
    if (m_min.empty() || m_min.top() >= value) { // min栈仅压入数据栈时的最小值(及其重复值)
        m_min.push(value);
    }
}
template<typename T>
void StackWithMin<T>::pop()
{
    if (m_data.empty()) {
        throw std::range_error("empty!");
    }
    if (m_min.top() == m_data.top()) { // 若数据栈出栈的是最小值，则min栈将其出栈
        m_min.pop();
    }
    m_data.pop();
}
template<typename T>
const T StackWithMin<T>::min() const
{
    if (m_data.empty()) {
        throw std::range_error("empty!");
    }
    return m_min.top();
}

int main()
{
    StackWithMin<int> s;
    s.push(3);
    assert(s.min() == 3);
    s.push(4);
    assert(s.min() == 3);
    s.push(2);
    assert(s.min() == 2);
    s.push(1);
    assert(s.min() == 1);
    s.push(2);
    s.push(1);
    s.push(3); // 3,4,2,1,2,1,3 || 3,2,1,1
    s.pop();
    assert(s.min() == 1);
    s.pop();
    s.pop();
    assert(s.min() == 1);
    s.pop();
    s.pop();
    s.pop();
    assert(s.min() == 3);
    s.pop();
    try {
        s.min();
    } catch (std::range_error) {
        cout << "[expected]std::range_error caught!" << endl;
    }
    cout << "OK!" << endl;
}