题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
解法
这题相对而言比较简单,栈的特点是先进后出
,队列的特点是先进先出
,因此用两个先进后出
的栈就能实现先进先出
的队列,在push
时直接往stack1
push,在pop
时,借助第二个栈暂存stack1
的其它元素,让stack1
栈底的元素(即最先进来的元素)出栈,出栈后不需要立刻将stack2
的元素弹回stack1
,因为如果下个操作还是pop
的话,就可以直接从stack2
弹出。
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
while (!stack2.isEmpty())
// s2不空,先将所有值转到s1
stack1.add(stack2.pop());
stack1.add(node);
}
public int pop() {
while (!stack1.isEmpty())
// s2为空
stack2.add(stack1.pop());
return stack2.pop();
}
}