剑指offer:2个栈实现队列
题目:用两个栈来实现一个队列,完成队列的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