题解 | #用两个栈实现队列#
用两个栈实现队列
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()就会报错。
查看9道真题和解析