题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。
解析
两个栈stack1 和stack2
入队列:直接压入stack1即可
出队列:如果stack2不为空,把stack2中的栈顶元素直接弹出;否则,把stack1的所有元素全部弹出压入stack2中,再弹出stack2的栈顶元素
实现
#include <iostream>
#include <stack>
using namespace std;
template<typename T> class MyQueue{
public:
MyQueue();
~MyQueue();
void addTail(const T& val); //在队列末尾添加元素
T deleteHead(); //删除队首元素
private:
stack<T> stack1;
stack<T> stack2;
};
template<typename T> MyQueue<T>::MyQueue(){
}
template<typename T> MyQueue<T>::~MyQueue(){
}
template<typename T> void MyQueue<T>::addTail(const T& val){
stack1.push(val);
}
template<typename T> T MyQueue<T>::deleteHead(){
if(stack2.empty()){
if(stack1.empty()){
throw new exception("队列是空的");
}else{
while(stack1.empty() == false){
stack2.push(stack1.top());
stack1.pop();
}
}
}
T head = stack2.top();
stack2.pop();
return head;
}
void Test(char actual, char expected)
{
if(actual == expected)
printf("Test passed.\n");
else
printf("Test failed.\n");
}
int main()
{
MyQueue<char> queue;
queue.addTail('a');
queue.addTail('b');
queue.addTail('c');
char head = queue.deleteHead();
Test(head, 'a');
head = queue.deleteHead();
Test(head, 'b');
queue.addTail('d');
head = queue.deleteHead();
Test(head, 'c');
queue.addTail('e');
head = queue.deleteHead();
Test(head, 'd');
head = queue.deleteHead();
Test(head, 'e');
system("pause");
return 0;
}