两个队列实现栈(Java)
分析思路:
现有两个队列 q1 和 q2,入栈则将元素加到 q1,出栈的时候先判读 q1 是否为空,因为 q1 中的元素总是后进来的,后进先出,除了队列的最后一个元素,将其它元素添加到 q2,q1 的最后一个元素出队。出栈的时候如果在 q2 中判断 q1 为空,除了 q2 的最后一个元素,将 q2 中其它元素添加到 q1,然后 q2 中的最后一个元素出队。
代码:
public class MyStack {
private Queue<Integer> q1 = new LinkedList<>();
private Queue<Integer> q2 = new LinkedList<>();
public MyStack() {
}
public boolean push(Integer num) {
return q1.offer(num);
}
public Integer pop() {
if (q1.isEmpty() && q2.isEmpty()) {
return null;
}
// 先判断 q1 是否为空
if (!q1.isEmpty()) {
int size = q1.size();
for (int i = 0; i < size - 1; i++) {
q2.offer(q1.poll());
}
return q1.poll();
} else {
int size = q2.size();
for (int i = 0; i < size - 1; i++) {
q1.offer(q2.poll());
}
return q2.poll();
}
}
public Integer size() {
return q1.size() + q2.size();
}
public static void main(String[] args) {
MyStack stack = new MyStack();
stack.push(1);
stack.push(2);
stack.push(3);
stack.push(4);
stack.push(5);
int size = stack.size();
for (int i = 0; i < size; i++) {
System.out.print(stack.pop() + " ");
}
System.out.println();
stack.push(6);
System.out.println(stack.pop());
}
}