- 用两个栈实现一个队列(P59)(面试题7)
- 两个队列实现一个栈
1.用两个栈实现一个队列(P59)(面试题7)
题目:用两个栈实现一个队列,队列的声明如下,请实现他的两个函数appendTail和deleteHead,分别完成在队列尾部插入结点和在队列头部删除结点的功能。
思路如下:
- 插入appendTail时,都压入栈1
- 删除时:判断栈2是否为空(先入先出FIFO)
- 若为空,栈1的元素都压入栈2,pop出栈2的栈顶就是队列的头;
- 若不为空可以直接pop栈2的栈顶元素
- 考虑异常,都为空时throw exception
#include<iostream>
#include<stack>
using namespace std;
template <typename T> class CQuene
{
public:
CQuene(){};
~CQuene(){};
void AppendTail(const T& iElement);
T DeleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
template <typename T>
void CQuene<T>::AppendTail(const T& iElement)
{
stack1.push(iElement);
}
template <typename T>
T CQuene<T>::DeleteHead()
{
if(stack2.empty())
{
if(stack1.empty())
throw exception("队列是空的");
else
{
while(!stack1.empty())
{
stack2.push(stack1.top());
stack1.pop();
}
}
}
T iHead = stack2.top();
stack2.pop();
return iHead;
}
void Test(char actual, char expected)
{
if(actual == expected)
printf("Test passed.\n");
else
printf("Test failed.\n");
try{
}
catch(exception& exception){
cout<<exception.what()<<endl;
}
}
int main()
{
CQuene<char> iquene;
iquene.AppendTail('a');
iquene.AppendTail('b');
iquene.AppendTail('c');
char head = iquene.DeleteHead();
Test(head , 'a');
head = iquene.DeleteHead();
Test(head , 'b');
iquene.AppendTail('d');
head = iquene.DeleteHead();
Test(head , 'c');
iquene.AppendTail('e');
head = iquene.DeleteHead();
Test(head , 'd');
head = iquene.DeleteHead();
Test(head,'e');
try{
iquene.DeleteHead();
}
catch(exception& exception){
cout<<exception.what()<<endl;
}
return 0;
}
测试用例:
- 往空的队列里添加、删除元素
- 往非空的队列里添加、删除元素
- 连续删除直至队列为空
2.两个队列实现一个栈的思路
思路如下:
- 插入appendTail时,往非空的队列里添加元素
- 删除时:判断哪个队列是非空的(先入后出FILO)
- 若都为空,throw exception
- 把非空的前n-1个元素都添加的空队列中,pop出第n个元素,也就是栈顶元素(队尾元素)
void StackPush(const int element)
{
if(!queue1.empty() && !queue2.empty())
throw expection("出错");
else if(queue1.empty() && !queue2.empty())
queue2.push(element);
else
queue1.push(element);
}
int StackPop()
{
if(!queue1.empty() && !queue2.empty())
throw expection("出错");
else if(queue1.empty() && queue2.empty())
throw expection("栈为空");
else if(queue1.empty() && !queue2.empty(){
while(queue2.size() > 1)
{
queue1.push(queue2.front());
queue2.pop();
}
return queue2.pop();
}
else{
while(queue1.size() > 1)
{
queue2.push(queue1.front());
queue1.pop();
}
return queue1.pop();
}
}