剑指offer:7-用两个栈实现队列

剑指offer:7-用两个栈实现队列_第1张图片

栈是常见的数据结构,在计算机领域广泛使用,如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;
}
运行结果如下:




你可能感兴趣的:(剑指offer:7-用两个栈实现队列)