栈是常见的数据结构,在计算机领域广泛使用,如OS会给每个线程创建一个栈用来存储函数调用时各个函数的参数、返回地址及临时变量等。特点是:后进先出。
删除操作:
stack2中元素弹出顺序刚好是stack1中元素进入顺序。故在删除元素时:
1>先判断stack2是否为空,若其不为空,则从中弹出元素;若为空,则将stack1中元素依次弹出并存入stack2中。
2>重复步骤1。
插入操作:
插入操作比较简单,直接插入stack1即可。
代码如下:
//剑指offer:7-用两个栈实现队列.
#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& node)
{
stack1.push(node);
}
//从队列中删除元素.
template <typename T>
T CQueue<T>::deleteHead()
{
//先判断stack2是否为空.
//若其不为空,则从中弹出元素;
//若为空,则将stack1中元素依次弹出并存入stack2中.
if(stack2.size() <= 0)
while (stack1.size() > 0)
{//当stack2为空时,将stack1中所有元素存入stack2中.
T &data = stack1.top();
stack1.pop();
stack2.push(data);
}
if(stack2.size() == 0)
throw new exception("queue is empty!");
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.\n" << 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;
}
运行结果如下: