IT公司100题-2-设计带min函数的stack

问题描述：

定义栈的数据结构，要求添加一个min 函数，能够得到栈的最小元素。

要求函数min、push 以及pop 的时间复杂度都是O(1)。

 

双倍空间实现：

保存2个栈，分别是元素和当前最小值。

 

压缩空间实现：

？

 // 2.cc
 #include <iostream>
 #include <stack>
 using namespace std;

 template <typename T> class MinStack {
 private:
       stack<T> stacks, mins;

 public:
       MinStack() {}
       ~MinStack() {}

       void push(const T & x) {
             stacks.push(x);
             if (mins.empty() || x <= mins.top())
                   mins.push(x);
       }

       void pop() {
             T x = stacks.top();
             stacks.pop();
             if (x == mins.top()) mins.pop();
             return;
       }

       T top() const{
             return stacks.top();
       }

       T get_min() {
             return mins.top();
       }
 };

 int main() {
       MinStack<int> s;
       s.push();
       s.push();
       s.push();
       s.push();
       s.push();
       cout << s.get_min() << endl;
       s.pop();
       s.pop();
       s.pop();
       cout << s.get_min() << endl;
       return ;
 }