题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
解题思路
使用栈来实现队列,首先要明白栈和队列的特点:栈是先进后出,队列是先进先出。
因此要用栈模拟队列,需要两个栈来操作。
stack1用于“入队”
stack2用于“出队”
假设stack1先入栈:1 2 3
此时:
stack1:{1, 2, 3}
stack2:{}
需要出队时,由于stack2为空,所以将stack1所有元素入栈到stack2,此时再从stack2中出栈。
两次先进后出,也就实现了先进先出。
此时:
stack1:{}
stack2:{3, 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() {
if (stack2.isEmpty()) {
while (stack1.size() != 0) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}