用两个栈实现一个队列的功能,用C++实现

 思路:

假设两个栈A和B,且都为空,

可以认为栈A提供入队列的功能,栈B提供出队列的功能,

入队列:入栈A

出队列:

>如果栈B不为空,直接弹出栈B的数据

>如果栈B为空,则依次弹出栈A的数据,放入栈B中,再弹出栈B的数据

代码如下:

#include <iostream>
#include <stack>

using namespace std;

template<typename T>
class MyQueue
{
public:
    void push(T t)
	{
		s1.push(t);
	}
	
	void pop()
	{
		if(s2.empty())
		{
			while(!s1.empty())
			{
				s2.push(s1.top());
				s1.pop();
			}
		}
		
		if(!s2.empty())
			s2.pop();
	}
	
	T front()
	{
		if(s2.empty())
		{
			if(s1.size() == 0)
				cout << "此队列为空" << endl;
			while(!s1.empty())
			{
				s2.push(s1.top());
				s1.pop();
			}
		}
		
		return s2.top();
	}
	
private:
    stack<T> s1;
	stack<T> s2;
};

int main()
{
    MyQueue<int> mq;
	mq.push(1);
	mq.push(2);
	mq.push(3);
	mq.push(4);
	mq.push(5);
	mq.push(6);
	mq.push(7);
	
	for(int i = 0;i < 7;i++)
	{
		cout << mq.front() <<endl;
		mq.pop();
	}
	
    return 0;
}

猜你喜欢

转载自blog.csdn.net/yue_jijun/article/details/81502334