题解 | #用两个栈实现队列#
用两个栈实现队列
https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6
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()){ return stack2.pop(); }else{ if(stack1.isEmpty()){ return -1; } while(!stack1.isEmpty()){ stack2.push(stack1.pop()); } } return stack2.pop(); } }
核心逻辑就是搞清楚什么时候才可以pop;队列是先进先出,栈是先进后出,
那么我们在pop的时候就要分两种状况:
1.当stack2不为空时,栈中的这些数据是之前留下的,也就是先加入的,根据先进先出,所以我们就可以直接将其pop()出来,
2.当stack2为空,stack1不为空时,那么stack1中的数据就都是后加入的数据,根据栈的先进后出,我们就需要将stack1中的数据全部遍历到stack2中取出第一个数据,该数据就是最先加入的数据了,当然如果stack1中没有数据,那么说明队列为空了,这时候pop()就会报错。