题目:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:栈的特点后进先出,队列的特点是先进先出。2个栈的情况可以通过2次的压栈和出栈实现队列的先进先出。
其中一个栈来实现入队,一个栈实现出队。
图示:
注意:对于入队操作,直接对stack1进行入栈操作就好了。
对于出队操作,需要有几个条件判断一下:
1.当stack1和stack2都为空的时候,出队操作是不合理的,必须进行判断
2.由于出队的操作是在stack2中进行的。首先stack2为空的时候,就需要先把stack1的元素弹出并压入到stack2中,stack2再弹出,这样就完成了出队操作。
当stack2不为空的时候,由于stack2的数据都是stack2出栈来的,顺序是一定的,所以直接弹出stack2栈顶元素就好。
import java.util.Stack;
public class StackToQueue {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
//入队:即为stack1的入栈
public void push(int node) {
stack1.push(node);
}
//出队
public int pop() {
//队列为空判断
if(stack1.empty()&&stack2.empty()){
throw new RuntimeException("队列为空");
}
//分stack2是否为空进行判断。
if(stack2.empty()){
while(!stack1.empty()){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}
github源码:栈实现队列操作
图片来源:https://cuijiahua.com/blog/2017/11/basis_5.html