用两个栈实现队列
分析:
首先,要清楚栈和队列的基础逻辑,栈是后进先出,队列是先进先出。如下图所示:
要用栈构建队列,两个栈SK1和SK2,栈每次弹出的都是新入的元素,而队列要求弹出最早进入的元素,
因此,可以将SK1中的元素 **倒入 ** SK2,弹出SK2栈顶的元素即可,
值得注意的是:对SK2非空的判断,当SK2为空时,就需要将SK1中的元素倒入,若不为空,就不做改变,如下图所示:
代码:
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() {
if(stack2.empty()){
while(stack1.size()!=0){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}