《剑指offer》面试题09：用两个栈实现一个队列

更多剑指offer面试习题请点击：《剑指offer》（第二版）题集目录索引

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

解题思路：
　　使用两个栈：s1、s2，所有的压栈Push操作全部在s1完成。关于出栈Pop操作则要分情况讨论：
　　1. 如果s2为空，则把s1中所有元素弹出，压到s2里面，然后弹出s2的栈顶元素即可。
　　2. 如果s2不为空，直接弹出s2的栈顶元素。

---

< code >

#ifndef __QS_H__
#define __QS_H__

#include <stack>
#include <exception>

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> CQueue<T>::CQueue(void)  //构造函数
{
}

template <typename T> CQueue<T>::~CQueue(void) //析构函数
{
}

template<typename T> void CQueue<T>::appendTail(const T& element) //压栈操作
{
    stack1.push(element);
}

template<typename T> T CQueue<T>::deleteHead() //出栈操作
{
    if (stack2.empty())  //如何stack2 为空， 就把stack1中的元素全部弹出压到 stack2 中。
    {
        while (stack1.size()>0)
        {
            T& data = stack1.top();
            stack1.pop();
            stack2.push(data);
        }
    }

    if (stack2.size() == 0) //走到这stack2还为空，则代表队列里面没有元素
        throw new exception("queue is empty");

    T head = stack2.top();
    stack2.pop();

    return head;
}
#endif;

< Test Code>

#include "QS.h"

void Test(char actual, char expected)
{
    if (actual == expected)
        printf("Test passed.\n");
    else
        printf("Test failed.\n");
}

int main(int argc, char* argv[])
{
    CQueue<char> queue;

    queue.appendTail('a');
    queue.appendTail('b');
    queue.appendTail('c');

    char head = queue.deleteHead();
    Test(head, 'a');

    head = queue.deleteHead();
    Test(head, 'b');

    queue.appendTail('d');
    head = queue.deleteHead();
    Test(head, 'c');

    queue.appendTail('e');
    head = queue.deleteHead();
    Test(head, 'd');

    head = queue.deleteHead();
    Test(head, 'e');

    system("pause");
    return 0;
}

测试结果：