[JAVA]两个队列实现栈
思路
首先,我们明白队列和栈的区别:队列是先进先出,栈是先进后出。所以我们可以发现,在代码逻辑上实现栈和队列的思想是相同的,只是实现的方式不同。所以用两个队列实现栈可以看作两个先进先出的链表实现的功能是一个先进后出的链表。
由此可知,两个队列中一定有一个队列元素的出队顺序是和出栈顺序相同的,所以我们可以用一个队列来实现栈的出栈操作,另一个队列来实现栈的入栈操作。因为队列是队尾入队头出,所以我们只要在一个空队列中先将待插入元素入队,再将现有元素入队,就可以实现与栈一样的效果。
具体图解
-
插入数字1到空队列A
-
插入数字2到队列B,再将队列A中元素依次出队,插入队列B
-
插入数字3到队列A,再将队列B中元素依次出队,插入队列A
源代码
import java.io.IOException;
import java.util.*;
/**
两个队列形成栈
*/
class Solution1{//初始化创建两个队列
Queue<Integer> A,B;
public Solution1() {
A = new LinkedList<>();
B = new LinkedList<>();
}
public void push(int x) {//保证一个队列为空,插入空队列再将另一个队列中数据插入此队列
if(!A.isEmpty() && B.isEmpty()){
B.add(x);
while(!A.isEmpty()){
B.add(A.poll());
}
}
if(A.isEmpty() && B.isEmpty()){
A.add(x);
}
}
public int pop() {//出栈,即将存有数据的非空队列输出一个数据
if(!A.isEmpty()){
return A.poll();
}
return B.poll();
}
public int top() {//获取栈顶元素,即获取存有数据的队列的队首数据且不删除
if(!A.isEmpty()){
return A.peek();
}
return B.peek();
}
public boolean empty() {//查看栈是否为空
if(A.isEmpty() && B.isEmpty()){
return true;
}
return false;
}
public static void main(String[] args){//主函数
Solution1 stack = new Solution1();
stack.push(1);
stack.push(2);
System.out.println(stack.top());
System.out.println(stack.pop());
System.out.println(stack.pop());
System.out.println(stack.empty());
}
}