careercup-栈与队列 3.5

3.5 实现一个MyQueue类，该类用两个栈来实现一个队列。

解答

队列是先进先出的数据结构(FIFO)，栈是先进后出的数据结构(FILO)， 用两个栈来实现队列的最简单方式是：进入队列则往第一个栈压栈， 出队列如果第二个栈不为空，则直接从第二个栈出队列，否则将第一个栈的数据依次压入第二个栈，然后出栈。每次有数据进入队列，直接进入第一个栈； 每次有数据出队列，在第二个栈不为空时直接从第二个栈出队。

C++实现代码：

#include<iostream>
#include<stack>
using namespace std;

class MyQueue
{
private:
    stack<int> s1;
    stack<int> s2;
public:
    void push(int x)
    {
        s1.push(x);
    }
    void pop()
    {
        if(s1.empty()&&s2.empty())
            return;
        if(s2.empty()&&!s1.empty())
        {
            while(!s1.empty())
            {
                s2.push(s1.top());
                s1.pop();
            }
            s2.pop();
        }
        else
            s2.pop();
    }
    int front()
    {
        if(s1.empty()&&s2.empty())
            return 0;
        if(!s2.empty())
            return s2.top();
        while(!s1.empty())
        {
            s2.push(s1.top());
            s1.pop();
        }
        return s2.top();
    }
    int size()
    {
        return s1.size()+s2.size();
    }
    bool empty()
    {
        return s1.empty()&&s2.empty();
    }
};

int main()
{
    MyQueue q;
    for(int i=0; i<10; ++i)
    {
        q.push(i);
    }
    cout<<q.front()<<endl;
    q.pop();
    q.push(10);
    cout<<q.front()<<endl;
    cout<<q.size()<<" "<<q.empty()<<endl;
    return 0;
}