题解 | #用两个栈实现队列#
用两个栈实现队列
https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6
逻辑大致如下:
·栈1 用来存储模拟队列push进来的数据,因为 栈是后进先出的,因此需要栈2配合一起使用
·当模拟队列需要pop数据出去,首先判断栈2是否为空,
·第一次肯定为空,因此会把栈1 的数据push到栈2,此时 栈2的数据最外面的是数据是在队列最前的。把最外面的一个数据pop出去。
每次模拟队列需要pop数据出去,都会判断栈2是否为空,空了才可以把栈1的数据迁过来。每次模拟队列push数据,其实都是存入栈1中。
import java.util.*; 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(stack2.isEmpty()){ while(!stack1.isEmpty()) { stack2.push(stack1.pop()); } } return stack2.pop(); } }