【剑指offer】面试题7----用两个栈实现队列

1、题目描述

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

template<typename T>
class CQueue{
public:
    CQueue(void);
    ~CQueue(void);

    void appendTail(const T& node);
    T deleteHead();

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

---

2、解题思路

首先由题目可得，该队列里面包含了两个栈，也就是用两个“先进后出”的栈来实现一个“先进先出”的队列。

我们可以这样来设计实现：

• 插入元素时，往stack1里面插入元素；
• 删除元素时，因为栈是“先进后出的”，所以先需要将stack1里的元素移到stack2里面去，再从stack2里面删除元素；
• 当stack1不为空，stack2为空时，就将stack1里的元素转移到stack2里面去；

如图所示：

---

3、源码实现

#include<iostream>
#include<stack>
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>
void CQueue<T>::appendTail(const T& node){
    stack1.push(node);
}

template<typename T>
T CQueue<T>::deleteHead(){
    //如果stack2为空，stack1不为空，将stack1中的元素转移到stack2中
    if (stack2.size() <= 0){
        while (stack1.size() > 0){
            T& data = stack1.top();
            stack1.pop();
            stack2.push(data);
        }
    }
    if (stack2.size() == 0){
        //说明两个栈都为空
        throw new exception("queue is empty");
    }

    //删除stack2里的元素
    T head = stack2.top();
    stack2.pop();

    return head;
}