题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入结点和在队列头部删除结点的功能。
思路:当stack2不为空时,在stack2中的栈顶元素是最先进入队列的元素,可以弹出;在插入一个元素时,直接将元素插入到stack1中。
核心代码如下:
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& element){ //往stack1插入节点
stack1.push(element);
}
template <typename T>
T CQueue<T>::deleteHead(){ //删除stack2的栈顶节点,若stack2为空,先将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");
T head = stack2.top();
stack2.pop();
return head; //返回删除的节点
}
拓展:用两个队列实现一个栈。也是根据队列的先进先出特性来模拟栈的后入先出,可通过画图的手段把抽象的问题形象化。