更多剑指offer面试习题请点击:《剑指offer》(第二版)题集目录索引
题目:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail
和 deleteHead
,分别完成在队列尾部插入结点和在队列头部删除结点的功能。
解题思路:
使用两个栈:s1、s2,所有的压栈Push操作全部在s1完成。关于出栈Pop操作则要分情况讨论:
1. 如果s2为空,则把s1中所有元素弹出,压到s2里面,然后弹出s2的栈顶元素即可。
2. 如果s2不为空,直接弹出s2的栈顶元素。
< code >
#ifndef __QS_H__
#define __QS_H__
#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.empty()) //如何stack2 为空, 就把stack1中的元素全部弹出压到 stack2 中。
{
while (stack1.size()>0)
{
T& data = stack1.top();
stack1.pop();
stack2.push(data);
}
}
if (stack2.size() == 0) //走到这stack2还为空,则代表队列里面没有元素
throw new exception("queue is empty");
T head = stack2.top();
stack2.pop();
return head;
}
#endif;
< Test Code>
#include "QS.h"
void Test(char actual, char expected)
{
if (actual == expected)
printf("Test passed.\n");
else
printf("Test failed.\n");
}
int main(int argc, char* argv[])
{
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');
system("pause");
return 0;
}
测试结果: