剑指Offer第21题（包含min函数的栈）

（本博客旨在个人总结回顾）

题目描述：

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

解题思路：

当题目思路不清晰时可使用画图或表格来模拟步骤，并慢慢发现其中的规律。

栈内压入3、4、2、1之后接连两次弹出栈顶数字再压入0时，数据栈、辅助栈和最小值状态

步骤	操作	数据栈	辅助栈	最小值
1	压入3	3	3,	3
2	压入4	3,4	3,3	3
3	压入2	3,4,2	3,3,2	2
4	压入1	3,4,2,1	3,3,2,1	1
5	弹出	3,4,2	3,3,2,	2
6	弹出	3,4	3,3	3
7	压入0	3,4,0	3,3,0	0

源码:

// ConsoleApplication1.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <iostream>
#include <stack>
using namespace std;

template <typename T>
struct StackWithMin
{
    void push(const T& value);
    void pop(const T& value);
    const T min();

    stack<T> m_data;
    stack<T> m_min;
    
};

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(const T& value)
{
    assert(m_data.size() > 0 && m_min.size() > 0);
    m_data.pop();
    m_min.pop();
}

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

int _tmain(int argc, _TCHAR* argv[])
{
    system("pause");
    return 0;
}