1、题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入节点和在队列头部删除节点的功能。
template<typename T>
class CQueue{
public:
CQueue(void);
~CQueue(void);
void appendTail(const T& node);
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
2、解题思路
首先由题目可得,该队列里面包含了两个栈,也就是用两个“先进后出”的栈来实现一个“先进先出”的队列。
我们可以这样来设计实现:
- 插入元素时,往stack1里面插入元素;
- 删除元素时,因为栈是“先进后出的”,所以先需要将stack1里的元素移到stack2里面去,再从stack2里面删除元素;
- 当stack1不为空,stack2为空时,就将stack1里的元素转移到stack2里面去;
如图所示:
3、源码实现
#include<iostream>
#include<stack>
using namespace std;
template<typename T>
class CQueue{
public:
CQueue(void);
~CQueue(void);
void appendTail(const T& node);
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
template<typename T>
void CQueue<T>::appendTail(const T& node){
stack1.push(node);
}
template<typename T>
T CQueue<T>::deleteHead(){
//如果stack2为空,stack1不为空,将stack1中的元素转移到stack2中
if (stack2.size() <= 0){
while (stack1.size() > 0){
T& data = stack1.top();
stack1.pop();
stack2.push(data);
}
}
if (stack2.size() == 0){
//说明两个栈都为空
throw new exception("queue is empty");
}
//删除stack2里的元素
T head = stack2.top();
stack2.pop();
return head;
}