7.用两个栈实现队列

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

思路：先将数据压入stack1中。删除时，把stack1所有的数据弹出到stack2压入，然后每次要删除第一个结点，就从stack2弹出；添加时，直接压入到stack1。

测试用例：
1.往空的队列里添加、删除元素。
2.往非空的队列里添加、删除元素
3.连续删除元素直至队列为空。

代码：

#include<iostream>
#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.size() <= 0)   //当stack2为空时，需将stack1所以元素弹出到stack2
    {
        while (stack1.size() > 0)
        {
            T& data = stack1.top(); //指向栈顶元素
            stack1.pop();
            stack2.push(data);
        }
    }

    if (stack2.size() == 0)
    {
        throw new exception();
    }

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

    return head;
}

void test(char actual, char expected)
{
    if (actual == expected)
    {
        cout << "Test Passed" << endl;
    }
    else
    {
        cout << "Test failed" << endl;
    }
}

int main()
{
    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');

    return 0;
}

相关题目：用两个队列实现一个栈。

思路：删除时，先从queue1中依次删除元素a、b并插入到queue2中，再从queue1中删除元素c。这就相当于从栈中弹出元素c。压入时，直接压入queue1。

7.用两个栈实现队列

猜你喜欢