题目:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
队列是先入先出,堆栈是先入后出,通过两个堆栈的push和pop操作可以实现堆栈的元素反转,从而达到先入先出的效果。
假设stack1是正常的栈,statck2是逆序的栈(和队列顺序一致),对于push操作,要保证新push的元素位于所有元素的最后,所以要先将stack2的元素pop出来push进stack1里,然后将新元素push,对于pop操作,要保证pop的元素的最先push进去的,相当于pop出栈底元素,所以要先将stack1的元素pop出来push进stack2,然后stack2pop出元素。
完整代码如下:
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
while(!stack2.empty()) {
int now = stack2.pop();
stack1.push(now);
}
stack1.push(node);
}
public int pop() {
while(!stack1.empty()) {
int now = stack1.pop();
stack2.push(now);
}
return stack2.pop();
}