题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead,分别完成在队列尾部插入结点和在队列头部删除结点的功能。
思路:先将数据压入stack1中。删除时,把stack1所有的数据弹出到stack2压入,然后每次要删除第一个结点,就从stack2弹出;添加时,直接压入到stack1。
测试用例:
1.往空的队列里添加、删除元素。
2.往非空的队列里添加、删除元素
3.连续删除元素直至队列为空。
代码:
#include<iostream>
#include<stack>
#include<exception>
using namespace std;
template <typename T> class CQueue
{
public :
CQueue(void);
~CQueue(void);
//在队列末尾添加一个结点
void appendTail(const T& node);
//删除队列的头结点
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
template<typename T> CQueue<T>::CQueue(void)
{
}
template<typename T> CQueue<T>::~CQueue(void)
{
}
template<typename T> void CQueue<T>::appendTail(const T& element)
{
stack1.push(element);
}
template<typename T> T CQueue<T>::deleteHead()
{
if (stack2.size() <= 0) //当stack2为空时,需将stack1所以元素弹出到stack2
{
while (stack1.size() > 0)
{
T& data = stack1.top(); //指向栈顶元素
stack1.pop();
stack2.push(data);
}
}
if (stack2.size() == 0)
{
throw new exception();
}
T head = stack2.top();
stack2.pop();
return head;
}
void test(char actual, char expected)
{
if (actual == expected)
{
cout << "Test Passed" << endl;
}
else
{
cout << "Test failed" << endl;
}
}
int main()
{
CQueue<char> queue;
queue.appendTail('a');
queue.appendTail('b');
queue.appendTail('c');
char head = queue.deleteHead();
test(head, 'a');
head = queue.deleteHead();
test(head, 'b');
queue.appendTail('d');
head = queue.deleteHead();
test(head, 'c');
queue.appendTail('e');
head = queue.deleteHead();
test(head, 'd');
head = queue.deleteHead();
test(head, 'e');
return 0;
}
相关题目:用两个队列实现一个栈。
思路:删除时,先从queue1中依次删除元素a、b并插入到queue2中,再从queue1中删除元素c。这就相当于从栈中弹出元素c。压入时,直接压入queue1。