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(6);
      s.push(7);
      s.push(5);
      s.push(9);
      s.push(3);
      cout << s.get_min() << endl;
      s.pop();
      s.pop();
      s.pop();
      cout << s.get_min() << endl;
      return 0;
}