使用兩個棧實現一個隊列,這樣對於stl中棧和隊列的使用算是一個小考察;
因爲棧的特性是後入先出,而隊列的特性是先入先出,所以就是最先入棧的數據要第一個出來,假設我們push的數據都放到了stack1中,所以我們pop的時候我們需要先將stack1中的數據都出到stack2中,這樣本來最後進來的數據就變成stack2中最先進來的數據,最先進來的數據變爲stack2中最後進來的數據,所以我們pop的時候直接pop stack2的就好;
代碼爲:
#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;
};