实现包含min函数的栈

要求实现一个带有min函数的堆栈，且push，pop以及min操作时间复杂度为O（1）。

思路：

在数据栈的基础上需要另一个辅助堆栈来存储当前最小元素的位置。新元素进栈时，将该元素与当前最小元素比较，若比当前最小元素还要小，则将新栈顶位置入辅助栈；否则仍然将原最小元素位置入辅助栈。出栈操作很简单，直接对两个栈进行pop操作。以下是该思路的C++实现，实现中采用了STL的deque容器，原因在于deque支持index操作。

 

#include <deque> #include <iostream> #include <stack> using namespace std; template<class T> class MinStack { public: void Push(const T&); void Pop() { min.pop(); data.pop_back(); } const T& Top() const { return data.back(); } const T& Min() const { return data.at(min.top()); } private: stack<int> min; deque<T> data; }; template<class T> void MinStack<T>::Push(const T& object) { if(min.empty()) min.push(0); else if(object < data.at(min.top())) min.push(data.size()); else min.push(min.top()); data.push_back(object); }