用两个栈实现一个队列功能的思想比较巧妙:
共分两步:
第一步,入队——用一个栈的push操作实现,将入队值压入一个栈中(栈一)
第二步,出队——在另一个栈中删除值(栈二):若栈二非空,直接从栈二中pop出即可
若栈二为空,则将栈一中的所有元素pop出来,一次push进栈二中,然后从栈二中pop出元素。
注:只有当stack2 为空时,才会从stack1中pop出元素push进stack2中。
以下程序来自程序员面试宝典(第三版) P179
#include <iostream>
#include <stack>
using namespace std;
template<class T>
struct MyQueue
{
void push(T &t)
{
s1.push(t);
}
T front()
{
if(s2.empty())
{
if(s1.size() == 0 ) throw;
while(!s1.empty())
{
s2.push(s1.top());
s1.pop();
}
}
return s2.top();
}
void pop()
{
if(s2.empty())
{
while(!s1.empty())
{
s2.push(s1.top());
s1.pop();
}
}
if(!s2.empty())
s2.pop();
}
stack<T> s1;
stack<T> s2;
};
int main()
{
MyQueue<int> mq;
int i;
for(i=0 ;i<10; ++i)
{
mq.push(i);
}
for(i=0; i<10; ++i)
{
cout<<mq.front()<<endl;
mq.pop();
}
return 0;
}