《剑指offer》面试题9:用两个栈实现队列

题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数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;
};

分析:当stack2 不为空时,在stack2 中的栈顶元素是最先进进入队列的元素,可以弹出。当stack2 为空时,我们把stack1 中的元素逐个弹出并压入stack2。由于先进入队列的元素被压到stack1 的底端,经过弹出和压入操作之后就处于stack2 的顶端,又可以直接弹出。
参考代码如下:

template<typename T> void CQueue<T>::appendTail(const T& element)
{
	stack1.push(element);
}
template<typename T> void 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;
}

测试用例:
a.往空的队列里添加、删除元素。
b.往非空的队列里添加、删除元素。
c.连续删除元素直至队列为空。

猜你喜欢

转载自blog.csdn.net/qq_43502142/article/details/84621593
今日推荐