剑指offer面试题（9）——用两个栈实现队列

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

        类的定义如下：

template <typename T> class CQueue
{
public:
    CQueue(void);
    ~CQueue(void);
    
    // 在队列末尾添加一个结点
    void appendTail(const T& node);

    // 删除队列的头结点
    T deleteHead();

public:
    stack<T> stack1;
    stack<T> stack2;
};

解题思路

        看到这道题，首先应该思考这两种数据结构有什么特点以及它们之间的联系。栈的特点是“先进后出”，队列的特点是“先进先出”，让我们用两个“先进后出”的栈来实现一个“先进先出”的队列，假如那么我们先把数据压入第一个栈，此时它的出栈顺序与想要实现的队列的出队顺序是相反的，如果再把这个栈里面的元素依次压入第二个栈，相当于对一组数据进行了两次倒序，此时第二个栈的出栈顺序就与数据出队的顺序一致。

        删除节点时，首先判断stack2是否为空，如果为空，则将stack1栈中的数据出栈并压入栈stack2中，操作完成后如果stack2依旧为空，则结束程序。

         上述解题步骤实现主要代码如下：

template<typename T> T CQueue<T>::deleteHead()
{
    if(stack2.empty())
    {
        while(!stack1.empty())
        {
            //T& data = stack1.top();
			stack2.push(stack1.top());
            stack1.pop();
            
        }
    }

    if(stack2.empty())
        throw new exception("queue is empty");

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

    return head;
}

        由于为模板类，所以声明和定义要放在同一个文件中。类的具体声明定义如下：

#pragma once
#include <stack>
#include <exception>

using namespace std;

template <typename T> class CQueue
{
public:
    CQueue(void);
    ~CQueue(void);
    
    // 在队列末尾添加一个结点
    void appendTail(const T& node);

    // 删除队列的头结点
    T deleteHead();

public:
    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())
    {
        while(!stack1.empty())
        {
            //T& data = stack1.top();
			stack2.push(stack1.top());
            stack1.pop();
            
        }
    }

    if(stack2.empty())
        throw new exception("queue is empty");

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

    return head;
}

测试用例1  

// ====================测试代码====================
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;
}