码迷,mamicode.com
首页 > 其他好文 > 详细

设计包含min函数的栈

时间:2014-12-17 12:21:21      阅读:166      评论:0      收藏:0      [点我收藏+]

标签: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;

}

  

设计包含min函数的栈

标签:blog   io   os   sp   on   数据   div   log   as   

原文地址:http://www.cnblogs.com/woshikafeidouha/p/4168861.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!