7. 用两个栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的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;
}