剑指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;
}