兩個棧實現一個隊列

使用兩個棧實現一個隊列,這樣對於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;
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章