《剑指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();
     s.Push();
     s.Push();
     s.Push();
     s.Push();
     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();
     cout<<s.Min()<<endl;
     s.Pop();
     cout<<s.Min()<<endl;
     s.Pop();
   //  cout<<s.Min()<<endl;
     s.Pop();
     s.Pop();
     return ;
 }