两个栈模拟实现一个队列

栈的特点是后进先出,队列的特点是先进先出,所以我们用两个栈来实现,我们需要分别实现进队列和出队列,stack1进栈,然后出队列,将stack1栈顶的元素全部进入stack2,然后stack出栈,就实现了先进先出。

 

#include

#include

using namespace std;

 

template

class CQueue

{

public:

    CQueue(){}

    ~CQueue(){}

    void insertTail(const T &val)

    {

        stack1.push(val);        //将元素插入stack1

    }

    T deleteHead()

    {

        if (stack2.size() <= 0)       //如果stack2为空

        {

            while (!stack1.empty())      //如果stack1不为空

            {

                stack2.push(stack1.top());    //如果队列不为空,将stack1中的元素逐个插入stack2

                stack1.pop();

            }

        }

        if (stack2.size() == 0)           //如果stack2为空,队列为空

            throw new exception("queue is empty");

 

        T head = stack2.top();

        stack2.pop();

        return head;

    }

private:

    stack stack1;

    stack stack2;

};

 

int main()

{

    CQueue queue;

    queue.insertTail('a');             //插入元素a

    queue.insertTail('b');             //插入元素b

    queue.insertTail('c');             //插入元素c

    cout << "第1次出队元素:" << queue.deleteHead() << endl;

    queue.insertTail('d');             //插入元素d

    cout << "第2次出队元素:" << queue.deleteHead() << endl;

    cout << "第3次出队元素:" << queue.deleteHead() << endl;

    cout << "第4次出队元素:" << queue.deleteHead() << endl;
    system("pause");
    return 0;

}

 

你可能感兴趣的:(算法题)