[牛客算法总结]： 用两个栈实现队列

标签：

栈、队列

题目：

用两个栈来实现一个队列，使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。

数据范围： n ≤ 1000
要求：存储n个元素的空间复杂度为 O(n)O(n) ，插入与删除的时间复杂度都是 O(1)O(1)

反思：

其实这道题目只要知道了栈和队列数据结构的话，用两个栈来实现队列的算法还是很容易写的。
主要解题思路：
我们push几个值进入tailStack例如 1、 2、 3，然后我们在push完毕之后再pop tailStack既进入了headStack就变成了 3 、 2、 1 ，这样子就实现了先进先出的效果了。
具体步骤：
1、我们定义两个栈 tailStack 和 headStack 。
2、 我们将push的值先push到tailStack中。
3、（主要实现）我们要pop的时候，我们先要判断headStack是否为空：
如果为空，我们使用while循环将tailStack中的值全部pop到headStack中，循环完毕之后再pop一个headStack的值。
如果不为空，我们直接pop一个headStack的值即可。

用到的知识点：

栈、队列

代码：

import java.util.EmptyStackException;
import java.util.Stack;

public class UseStackImplementQueue {
	Stack<Integer> tailStack = new Stack<Integer>();
	Stack<Integer> headStack = new Stack<Integer>();

	public void push(int node) {
		tailStack.push(node);
	}

	public int pop() {
		if (headStack.empty()) {
			while (!tailStack.empty()) {
				headStack.push(tailStack.pop());
			}
			return headStack.pop();
		} else {
			return headStack.pop();
		}
	}
}