《剑指offer》面试题21—包含min函数的栈

题目：定义栈数据结构，并在该数据结构中实现一个能获得栈最小元素的函数min。要求push，min，pop时间都是O(1)。

思路：要用一个辅助栈，每次有新元素压栈时辅助栈压入当前最小元素；min函数直接取辅助栈栈顶元素即可；有元素弹出时辅助栈元素也弹出，这样栈顶就是剩下的元素中的最小的了。

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

template<typename T>
class StackWithMin
{
public:
    void Push(const T& element);
    T Min();
    void Pop();

private:
    stack<T> stack_data;
    stack<T> stack_min;
};

template<typename T> void StackWithMin<T>::Push(const T& element)  //类后面的那个T太容易忘了。。
{
    if(stack_data.empty())
    {
        stack_data.push(element);
        stack_min.push(element);
    }
    else
    {
        stack_data.push(element);
        T temp = stack_min.top();
        if(element < temp)
            stack_min.push(element);
        else
            stack_min.push(temp);
    }
}

template <typename T> T StackWithMin<T>::Min()
{
    if(stack_data.empty())      //这里一开始忘记检查是否为空了！
    {
        cout<<"Empty Stack!"<<endl;
        //return;   这里应该要assert一下
    }
    return stack_min.top();
}

template <typename T> void StackWithMin<T>::Pop()
{
    if(stack_data.empty())      //这里一开始忘记检查是否为空了！
    {
        cout<<"Empty Stack!"<<endl;
        return;
    }
    stack_data.pop();
    stack_min.pop();
}


int main()
{
    StackWithMin<int> s;
    s.Push(3);
    s.Push(3);
    s.Push(4);
    s.Push(2);
    s.Push(1);
    cout<<s.Min()<<endl;
    s.Pop();
    cout<<s.Min()<<endl;
    s.Pop();
    cout<<s.Min()<<endl;
    s.Pop();
    cout<<s.Min()<<endl;
    s.Pop();
    s.Push(0);
    cout<<s.Min()<<endl;
    s.Pop();
    cout<<s.Min()<<endl;
    s.Pop();
  //  cout<<s.Min()<<endl;
    s.Pop();
    s.Pop();
    return 0;
}