首页 > 代码库 > 设计包含min函数的栈

设计包含min函数的栈

定义栈的数据结构,要求添加一个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;}

  

设计包含min函数的栈