剑指offer 面试题7

题目：

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数appendTail和deleteHead，分别完成在队列尾部插入节点和队列头部删除节点的功能。

在解这道题的时候，直接想很难想出各种情况。举个例子，思路就清晰了。

代码如下：

#include<iostream>
#include<stack>
using namespace std;

template<typename T>
class CQueue
{
public:
void appendTail(const T&node);
T deleteHead();
private:
stack<T>stack1;
stack<T>stack2;

};

template<typename T>
void CQueue<T>::appendTail(const T&node)
{
stack2.push(node);
}

template<typename T>
T CQueue<T>::deleteHead()
{
if (stack1.size() == 0)
{
if (stack2.size() > 0)
{
while (stack2.size() > 0)
{
T& element = stack2.top();
stack1.push(element);
stack2.pop();
}
}
else
throw exception("queue is empty!");
}
T &head = stack1.top();
stack1.pop();
return head;
}

int main()
{
CQueue<int> q;
q.appendTail(1);
q.appendTail(2);
q.appendTail(3);
cout << q.deleteHead()<<" ";
q.appendTail(4);
cout << q.deleteHead() << " ";

return 0;
}