用两个栈实现一个队列

栈是先进后出,队列是先进先出

每次元素都push在st1中,pop的时候如果st2为空,将st1的栈顶元素放在st2的栈底,这样st1的所有元素都放在st2中,st1的栈底就是st2的栈顶,pop st2的栈顶,这样就满足了队列的先进先出。
用两个栈实现一个队列_第1张图片

#include 
using namespace std;
#include 
#include 

template <class T>
class SQueue {
public:
    void Push(const T& value);
    T Pop();
private:
    stack  st1;
    stack  st2;
};

template <class T>
T SQueue::Pop()
{
    if (st2.size() <= 0)
    {
        if (st1.size() == 0)
        {
            exit(1);
        }
        while ((st1.size() > 0))
        {
            T& top = st1.top();
            st2.push(top);
            st1.pop();
        }
    }

    T head = st2.top();
    st2.pop();
    return head;

}

template <class T>
void SQueue::Push(const T& value)
{
    st1.push(value);
}

int main()
{
    SQueue<int> sq;
    for (int i = 0; i < 10; ++i)
    {
        sq.Push(i);
    }
    for (int i = 0; i < 5; ++i)  
    {
        cout << sq.Pop() << " ";
    }

    for (int i = 0; i < 5; ++i)  //分两次验证
    {
        cout << sq.Pop() << " ";
    }
    cout << endl;

    system("pause");
    return 0;
}

这里写图片描述

你可能感兴趣的:(剑指offer)