使用两个栈实现一个队列,这样对于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;
};