标签:blog io os sp on 数据 div log as
定义栈的数据结构,要求添加一个min函数,能够得到栈的最小元素。要求函数min、pop、push的时间复杂度为O(1);
方法:采用一个辅助栈,该栈用来存放最小元素。当进行push操作时,将栈中的最小元素同时push进辅助栈中,当pop操作时同时pop掉辅助栈中的栈顶元素。这样元素栈对应位置的辅助栈中存放的永远是最小元素。
#include<iostream> #include<stack> using namespace std; template<class T> class StackWithMin { public: void push(const T value); void pop(); T min(); private: stack<T> dataStack; stack<T> minStack; }; template <class T> void StackWithMin<T>::push(T value) { dataStack.push(value); if(minStack.size()==0) { minStack.push(value); } else if(value<minStack.top()) { minStack.push(value); } else { minStack.push(minStack.top()); } } template <class T> void StackWithMin<T>::pop() { dataStack.pop(); minStack.pop(); } template <class T> T StackWithMin<T>::min() { T p; p=minStack.top(); return p; } int main() { int t; StackWithMin<int> s; s.push(3); s.push(1); s.push(5); s.push(6); s.push(0); t=s.min(); cout<<"min:"<<t<<endl; s.pop(); t=s.min(); cout<<"min:"<<t<<endl; return 0; }
标签:blog io os sp on 数据 div log as
原文地址:http://www.cnblogs.com/woshikafeidouha/p/4168861.html