剑指offer-包含min函数的矩阵

/*******************************************************************
Copyright(c) 2016, Tyrone Li
All rights reserved.
*******************************************************************/
// 作者:TyroneLi
//

/*
Q:
    包含min函数的矩阵：
        定义栈的数据结构，请在该类型中实现一个可以得到栈的最小
        元素的min函数，在该栈中，调用min,pop,push的时间复杂度都是O(1).
S:
    定义一个辅助栈，原栈中每压入一个元素，都在辅助栈中压入新的最小在原栈中的最小值元素。
*/

#include <iostream>
#include <stack>

#include <cstdio>
#include <cassert>
#include <cstdlib>
#include <cstring>

template<typename T>
class StackWithMin
{
public:
    StackWithMin() {}
    ~StackWithMin() {}

    T&top();
    const T&top() const;

    void push(const T&value);
    void pop();

    const T&min() const;

    bool empty() const;
    size_t size() const;

private:
    std::stack<T> m_data;
    std::stack<T> m_min;
};

template<typename T>
T&StackWithMin<T>::top()
{
    assert(m_data.size() > 0);
    return m_data.top();
}

template<typename T>
const T&StackWithMin<T>::top() const
{
    assert(m_data.size() > 0);
    return m_data.top();
}

template<typename T>
void StackWithMin<T>::push(const T&value)
{
    // 把新元素先添加到原栈
    m_data.push(value);

    // 当新元素比之前的最小元素小时，把新元素插入辅助栈里；
    // 否则把之前的最小元素重复插入辅助栈里
    if(m_min.size() == 0 || m_min.top() > value)
        m_min.push(value);
    else
        m_min.push(m_min.top());
}

template<typename T>
void StackWithMin<T>::pop()
{
    assert(m_data.size() > 0 && m_min.size() > 0);

    m_data.pop();
    m_min.pop();
}

template<typename T>
const T&StackWithMin<T>::min() const
{
    assert(m_min.size() > 0);

    return m_min.top();
}

template <typename T> 
bool StackWithMin<T>::empty() const
{
    return m_data.empty();
}

template <typename T> 
size_t StackWithMin<T>::size() const
{
    return m_data.size();
}

void Test(const char* testName, const StackWithMin<int>& stack, int expected)
{
    if(testName != nullptr)
        printf("%s begins: ", testName);

    if(stack.min() == expected)
        printf("Passed.\n");
    else
        printf("Failed.\n");
}

int main(int argc, char* argv[])
{
    StackWithMin<int> stack;

    stack.push(3);
    Test("Test1", stack, 3);

    stack.push(4);
    Test("Test2", stack, 3);

    stack.push(2);
    Test("Test3", stack, 2);

    stack.push(3);
    Test("Test4", stack, 2);

    stack.pop();
    Test("Test5", stack, 2);

    stack.pop();
    Test("Test6", stack, 3);

    stack.pop();
    Test("Test7", stack, 3);

    stack.push(0);
    Test("Test8", stack, 0);

    return 0;
}