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

两个栈实现一个队。

时间:2016-05-02 17:11:45      阅读:177      评论:0      收藏:0      [点我收藏+]

标签:两个栈实现一个队。

思路:

入队:

只将栈s1作为入队。


出队:

将栈s2作为出队,如果栈s1不为空,s2为空,出队时将s1的元素倒入s2中,再pop();

否则s2不为空,出队时直接在栈s2中pop();

#include<iostream>
#include<stack>
#include<assert.h>
using namespace std;
template<class T>
class	Queue
{
public:
	Queue()
		:_size(0)
	{}
	void appendTail(T num)
	{
		s1.push(num);
		_size++;
	}
	void deleteHead()
	{
		assert(_size>0);
		if(s2.empty())
		{
			while(s1.size()!=1)
			{
				s2.push(s1.top());
				s1.pop();
			}
			s1.pop();
		}
		else
		{
			s2.pop();
		}
		_size--;
	}
	T& top()
	{
		assert(_size>0);
		if(s2.empty())
		{
			while(s1.size()!=1)
			{
			   s2.push(s1.top());
			}
			return s1.top();
		}
		else
		{
			return s2.top();
		}
	}
private:
	stack<T> s1;
	stack<T> s2;
	int _size;
};
int main()
{
	Queue<int> q;
	q.appendTail(1);
	q.appendTail(2);
	q.appendTail(3);
	q.appendTail(4);
	q.deleteHead();
	q.deleteHead();
	q.deleteHead();
	q.deleteHead();
	q.appendTail(5);
	cout<<q.top()<<endl;
	system("pause");
	return 0;
}


本文出自 “liveyoung” 博客,转载请与作者联系!

两个栈实现一个队。

标签:两个栈实现一个队。

原文地址:http://10707042.blog.51cto.com/10697042/1769453

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