上一篇文章我讲述了用两个栈实现一个队列,这一篇讲述用两个队列实现一个栈。
题目:用两个栈实现一个队列,实现两个函数appendTop和deleteHead,分别完成在栈顶插入元素和弹出一个栈顶元素。
实现过程:将queue1作为主队列,每次将插入的数据放在queue1的队尾。当执行deleteTop时,先判断queue1是否为空,若为空,则返回错误信息。否则,将queue1中除了队尾的元素全都依次插入到队列queue2中,将queue2作为一个临时队列,然后输出queue1的队尾元素,即为head的值。最后再将queue2中的元素重新填写到queue1中。
实现代码如下:
#include<iostream>
#include<queue>
using namespace std;
class CStack
{
public:
CStack(void){};
~CStack(void){};
void appendTop(const int &data)
{
queue1.push(data);
}
int deleteTop()
{
int head=-1;
if(queue1.empty())
{
cout<<"stack is empty,head =";
return head;
}
else
{
while(queue1.size()!=1)
{
queue2.push(queue1.front());
queue1.pop();
}
head=queue1.front();
queue1.pop();
while(queue2.size()>0)
{
queue1.push(queue2.front());
queue2.pop();
}
}
return head;
}
private:
queue<int> queue1;
queue<int> queue2;
};
int main()
{
CStack stack;
stack.appendTop(1);
stack.appendTop(2);
cout<<stack.deleteTop()<<endl;
stack.appendTop(3);
cout<<stack.deleteTop()<<endl;
cout<<stack.deleteTop()<<endl;
cout<<stack.deleteTop()<<endl;
return 0;
}