题目
用两个栈来实现一个队列,完成队列的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();
}
}