用两个栈实现队列

Posted by DH on August 4, 2017

题目

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

分析

栈的特点是“先进后出”,队列的特点是“先进先出”。 在插入的时候,先将第一个元素压入栈的栈底,后面的元素依次压入栈顶,最后一个压入的元素在栈顶。 而出栈的时候,必须从栈顶开始出栈,也就是最先入栈的元素最后出栈。

这时,我们将所有元素{1,2,3,4,5}压入栈1,从栈底到栈顶,依次是{1,2,3,4,5}。栈1的出栈顺序是{5,4,3,2,1}。每次,在栈1出栈一个元素后, 就将元素压栈到栈2,这时,栈2从栈底到栈顶的元素是{5,4,3,2,1},那么栈2出栈的时候,顺序正好是{1,2,3,4,5}。这样就实现了先进先出的队列的要求。

进一步分析,

将元素{3,2,1}压入栈2,并且弹出元素1和2,此时再将4压入栈1,如图所示:

在这种情况下,如果我们将4从栈1pop,并且push到栈2,就会出现问题,此时会先弹出4。但是只要我们每次保持栈2为空的时候,才从栈1pop出元素, 并且push到栈2,就不会有问题。

代码

import java.util.Stack;

public class Solution {
    // 栈1用于入栈
	Stack<Integer> stack1 = new Stack<>();
	// 栈2用于出栈
	Stack<Integer>stack2 = new Stack<>();
	
	public void push(int node) {
		stack1.push(node);
	}
	
	public int pop() {
		// 每次都从栈2弹出,没有元素就从栈1弹出,并存入栈2
		if (stack2.isEmpty()) {
			while(!stack1.isEmpty())
			stack2.push(stack1.pop());
		}
		
		return stack2.pop();
	}
}