题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
栈是先进后出的数据结构,而队列是先进先出。为了实现队列,那么我们必须将数据先压入其中一个栈(栈A)中,数据压入完毕之后,将数据都出栈,并压入另一个栈(栈B)中,压入完毕之后栈B中元素出栈,就达到了先进先出的效果。
那么我们在push时所需要做的工作就是把数据压入栈A
在pop时,首先判断两个栈是否为空,如果都为空则抛出异常。如果栈B不为空,那么栈B元素出栈,如果栈B为空,将栈A元素都出栈并压入栈B中,压入完毕之后,将栈B元素出栈。
代码
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(stack1.empty() && stack2.empty()){
throw new RuntimeException("Queue is empty");
}
//如果栈B为空,那么将栈A的元素都出栈并压入栈B
if(stack2.empty()){
while(!stack1.empty()){
int t = stack1.pop();
stack2.push(t);
}
}
//栈B元素一一出栈
return stack2.pop();
}
}
拓展:两个队列实现栈
用两个队列实现栈更困难一些,因为队列是先进先出,你会发现如果把数据一起操作,无论怎样都是先进先出的顺序。例如:将1、2、3分别入队(队A),再出队,然后压入另一个队列(队B),出队顺序还是1、2、3。所以我们必须要把数据分开操作。对于刚刚例子中,我们期望的出队数字为3,所以我们将1、2从队A出队,并放入队B中,然后队A还剩下3,我们将3出队并存入变量c中,再将1、2出队B,放回队A中。返回值为c。简单来说每次pop时,都将队A元素出队,直到只剩下一个元素为止。
import java.util.PriorityQueue;
import java.util.Queue;
class Solution {
Queue<Integer> q1 = new PriorityQueue<Integer>();
Queue<Integer> q2 = new PriorityQueue<Integer>();
public void push(int node) {
q1.offer(node);
}
public int pop() {
//如果队列都为空,抛出异常
if(q1.isEmpty() && q2.isEmpty()){
throw new RuntimeException("Stack is empty");
}
int t=0;
//如果队A中只有一个元素,那么出队
if(q1.size()==1)t = q1.poll();
//如果队A中不止一个元素,那么将元素出队并压入队B,直到队A只剩下一个元素
else if(q1.size() > 1){
while(q1.size() > 1){
q2.offer(q1.poll());
}
//将队A中剩下的元素出队并存入t中
t = q1.poll();
//将其他元素再放回队A中,以便下次使用。将队B中元素出队并压入队A
while(!q2.isEmpty()){
q1.offer(q2.poll());
}
}
return t;
}
}