题目:用两个栈实现一个队列,实现头删和尾插的功能。
思路:栈是先进后出,队列是先进先出。用两个栈InsertStack和TmpStack。如果要插入数据就只往InsertStack中插。删除数据时,如果TmpStack为空,则将InsertStack中的数据逐个拿出来,插入到TmpStack中,这样先入InsertStack栈的元素,就被翻转到TmpStack的栈顶,就可以在TmpStack中直接删除元素。
#include<iostream>
#include<stack>
using namespace std;
template<typename T>
class Queue
{
public:
Queue(){}
~Queue(){}
void PushTail(const T& value);
void PopHead();
private:
stack<T> InsertStack;
stack<T> TmpStack;
};
template<typename T> void Queue<T>::PushTail(const T& value)
{
InsertStack.push(value);
cout << "Insert " << value << " success" << endl;
}
template<typename T> void Queue<T>::PopHead()
{
if (TmpStack.empty())
{
while (!InsertStack.empty())
{
T tmp = InsertStack.top();
TmpStack.push(tmp);
InsertStack.pop();
}
}
if (TmpStack.size() == 0)
{
cout << "Queue is empty" << endl;
return ;
}
T PopValue = TmpStack.top();
TmpStack.pop();
cout << "Pop value is " << PopValue << endl;
}
void Test()
{
Queue<int> q;
q.PushTail(1);
q.PushTail(2);
q.PushTail(3);
q.PopHead();
q.PopHead();
q.PopHead();
q.PopHead();
}
int main()
{
Test();
getchar();
return 0;
}