《剑指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;
}

思路：

从上述队列的声明中可以看出，一个队列包含了两个栈stack1和stack2，因此这道题的意图是要求我们操作这两个”先进后出“的栈实现一个”先进先出“的队列CQueue.

步骤：

1.stack1用来保存入队列数据.
2.当stack2不为空时，在stack2中的栈顶元素是最先进入队列的元素，可以弹出。当stack2为空时，我们把stack1中的元素逐个弹出并压入stack2。由于先进入队列的元素被压到stack1的底端，经过弹出和压入操作之后就处于stack2的顶端，又可以直接弹出。

代码：

package test;

import java.util.Stack;

public class CQueue<T> {

	private Stack<T> stack1 = new Stack<>();
	private Stack<T> stack2 = new Stack<>();
	
	//队列添加元素
	public void appendTail(T t) {
		stack1.push(t);
	}
	
	//队列删除元素
	public T deleteHead() {
		if(stack2.size() <= 0) {
			while(stack1.size() > 0) {
				T data = stack1.pop();
				stack2.push(data);
			}
		}

		return stack2.pop();
	}
	
	public static void main(String[] args) {
		CQueue<Integer> cQueue = new CQueue<>();
		cQueue.appendTail(1);
		cQueue.appendTail(2);
		cQueue.appendTail(3);
		System.out.println(cQueue.deleteHead());
	}
}