《剑指offer》读书笔记
题目:
用两个栈实现一个队列,队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入结点
和队列头部删除结点的功能。
template <typename T>class CQueue
{
public:
CQueue(void);
~CQueue(void);
void appendTail(const T& node);
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
思路:在纸上画画,怎样用两个栈子实现一个队列的功能呢?其实很容易就能发现规律了,假设有a,b,c三个元素,将它们一个一个按序压入栈中,假设有stack1和stack2两个栈,先将其压如stack1,得{c,b,a},a最先压入,只能在栈底,现在再将stack1中的元素一次pop(),压入stack2中,得到{}a,b,c},这时序列的顺序就和队列一模一样了。appendTail()只要对stack1进行push()就好了,deleteHead()只要对stack2进行pop()即可。
下面是完整的实现及测试代码:
// TwoStackToQueue.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include<iostream>
#include<stack>
#include<queue>
#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(){}
template<typename T>
CQueue<T>::~CQueue(){}
template<typename T>
void CQueue<T>::appendTail(const T& node)
{
stack1.push(node);
}
template<typename T>
T CQueue<T>::deleteHead()
{
if (stack2.size() <= 0)
{
while (stack1.size() != 0)
{
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;
}
int main()
{
CQueue<char> queue1;
queue1.appendTail('a');
queue1.appendTail('b');
queue1.deleteHead();
queue1.deleteHead();
char testch = queue1.deleteHead();
cout << "testch == " << testch << endl;
return 0;
}