兩個棧實現一個隊列，兩個棧實現隊列

兩個棧實現一個隊列，兩個棧實現隊列

用兩個棧實現一個隊列的功能。

解題思路

假設兩個棧A和B，且都為空白。
棧A提供push()功能，棧B提供pop()功能。

• 入隊列：入棧A。
• 出隊列：
  
  • 如果棧B不為空白，直接彈出B的元素。
  • 如果棧B為空白，則依次彈出棧A的元素並壓入棧B中，再彈出B中的元素。

實現代碼

#include <iostream>#include <stack>using namespace std;template<class T>class MyQueue{public:    void push(const T& t)    {        s1.push(t);    }    void pop()    {        if (s2.empty())        {            while (!s1.empty())            {                s2.push(s1.top());                s1.pop();            }        }        s2.pop();    }    T& top()    {        if (s2.empty())        {            while (!s1.empty())            {                s2.push(s1.top());                s1.pop();            }        }        return s2.top();    }    bool empty() const    {        return s1.empty() && s2.empty();    }private:    stack<T> s1;    stack<T> s2;};int main(){    MyQueue<int> myque;    for (int i = 0; i < 10; i++)    {        myque.push(i);    }    while (!myque.empty())    {        cout<<myque.top()<<" ";        myque.pop();    }}

注意：pop()和top()中沒有再加上if (!s2.empty())的判斷，我認為此處應該由使用者來判斷MyQueue對象是否為空白，減少一些不必要的判斷可以提高程式效率。

著作權聲明：本文為博主原創文章，未經博主允許不得轉載。