刷题这些天,每天都靠看题解过日子,今天又随机了一道题,用两个栈实现队列,看题目感觉挺不错的,但是看输入输出的样例突然就懵了,我感觉这道题可能就是题目看着比较麻烦其实做起来不麻烦,下面是题:
题目链接:用两个栈实现队列
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:
输入:
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出:[null,null,3,-1]
示例 2:
输入:
[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]
提示:
1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用
思路也很简单,就是两个栈,a栈就用来存放入队的东西,出队的时候,把a栈里的东西都依次弹出,压入b栈,然后b栈再出栈,就是先了出队,也就是实现了先进先出,那么出队一个数的话,再次入队还是往a栈里面放,当b出完之后(为空),再把a栈里面的依次弹出压入b栈。
总的来说的就是,a栈用来放入队的值,b栈用来放出队的值,b栈为空的时候把a栈里面的值依次弹出压入b栈即可。
下面是代码,也是纪念以下自己瞬间就有思路的题吧…
import java.util.Stack;
public class CQueue {
Stack<Integer> a;
Stack<Integer> b;
public CQueue() {
// 找两个栈,一个栈存数据,另一个栈作出队操作
a = new Stack<>();
b = new Stack<>();
}
public void appendTail(int value) {
// 入队操作
a.push(value);
}
public int deleteHead() {
if (b.empty()){
int size = a.size();
// b栈为空,则把a里面东西放里头操作
for (int i = 0; i < size; i++){
b.push(a.pop());
}
}
try{
Integer pop = b.pop();
if (pop != null) return pop;
}catch (Exception e){
return -1;
}
return -1;
}
}