数据结构与算法之使用两个栈实现一个队列

使用堆栈实现队列

原创于 2020-01-01 18:24:56 发布 · 197 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法同时被 2 个专栏收录

30 篇文章

订阅专栏

面试指南

25 篇文章

订阅专栏

本文介绍了一种使用两个堆栈来实现队列数据结构的方法。通过将元素压入一个堆栈并在另一个堆栈中弹出，可以实现队列的先进先出特性。文章提供了完整的Java代码示例，展示了如何在不同堆栈之间转移元素以维持队列的性质。

public class QueueOfStack<T> {

	private Stack<T> stackA = null;
	private Stack<T> stackB = null;

	public QueueOfStack() {
		stackA = new Stack<T>();
		stackB = new Stack<T>();
	}

	/**
	 * 入队
	 * 
	 * @param obj
	 */
	public synchronized void inQueue(T obj) {
		stackA.push(obj);
	}

	/**
	 * 出队
	 * 
	 * @return
	 */
	public synchronized Object outQueue() {
		if (stackA.isEmpty())
			return null;
		while (!stackA.isEmpty())
			stackB.push(stackA.pop());
		T t = stackB.pop();
		while (!stackB.isEmpty()) {
			stackA.push(stackB.pop());
		}
		return t;
	}

	/**
	 * 出队
	 * 
	 * @return
	 */
	public synchronized T outQueueV2() {
		if (stackB.isEmpty()) {
			if (stackA.isEmpty())
				return null;
			while (!stackA.isEmpty()) {
				stackB.push(stackA.pop());
			}
		}
		return stackB.pop();
	}

	public static void main(String[] args) {
		QueueOfStack<String> queue = new QueueOfStack<String>();
		queue.inQueue("1");
		queue.inQueue("2");
		queue.inQueue("3");
		System.out.println(queue.outQueueV2());
		System.out.println(queue.outQueueV2());
		System.out.println(queue.outQueueV2());
	}

}