栈的特点是“先进后出”,队列的特点是“先进先出”。
再插入的时候,先将第一个元素压入栈的栈底,后面的元素依次压入栈顶,最后一个压入的元素在栈顶。
而出栈的时候,必须从栈顶开始出栈,也就是最先入栈的元素最后出栈。
这时,我们将所有元素{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 {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
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();
}
}