两个栈实现一个队列

使用两个栈实现一个队列,这样对于stl中栈和队列的使用算是一个小考察;

因为栈的特性是后入先出,而队列的特性是先入先出,所以就是最先入栈的数据要第一个出来,假设我们push的数据都放到了stack1中,所以我们pop的时候我们需要先将stack1中的数据都出到stack2中,这样本来最后进来的数据就变成stack2中最先进来的数据,最先进来的数据变为stack2中最后进来的数据,所以我们pop的时候直接pop stack2的就好;

剑指offer

代码为:

#include <iostream>
#include <stack>
#include <assert.h>
using namespace std;


template <class T>
class SQueue
{
public:
    void Push(const T& x)
    {
        s1.push(x);//直接放入s1中
    }
    void Pop()
    {
        if (s2.size() == 0)
        {
            int count = s1.size();
            int tmp = 0;
            while (count != 0)//如果s2为空的话,就将s1的数据全部移到s2中
            {
                tmp = s1.top();
                s1.pop();
                s2.push(tmp);

                count--;
            }
        }

        if (s2.size() == 0)
        {
            return;//说明没有数据
        }
        s2.pop();
    }
    T& Front()
    {
        if (s2.size() != 0)
        {
            return s2.top();
        }
        else if (s1.size() != 0)
        {
            int count = s1.size();
            int tmp = 0;
            while (count != 0)
            {
                tmp = s1.top();
                s1.pop();
                s2.push(tmp);

                count--;
            }

            return s2.top();
        }
        else
        {
            cout << "队列为空"<<endl;
            assert(-1);
        }

    }

    size_t Size()
    {
        return s1.size() + s2.size();
    }
    bool Empty() const
    {
        return s1.empty() || s2.empty();
    }
protected:
    stack<T> s1;
    stack<T> s2;
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章