用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目如上,出处leetcode官网
由于我是Android开发的原因,最开始采用的是kotlin语言编写
class CQueue() {
//新建一个输入栈
val s = Stack<Int>()
//一个输出栈
val s2 = Stack<Int>()
fun appendTail(value: Int) {
if (s.empty()) {
while (!s2.empty()) {
s.add(s2.pop())
}
}
s.add(value)
}
fun deleteHead(): Int {
//如果输出栈没有内容,则将输入栈反转并压入输出栈,然后pop即可达到队头删除的要求
if (s2.empty()) {
while (!s.empty()) {
s2.add(s.pop())
}
}
if (s2.empty()){
return -1
}
return s2.pop()
}
}
无疑这个写法他没毛病,但是当我查看官方解体之后发现3处可优化的地方
1、在appendTail
函数执行时可以直接添加,不需要将s2中的数据反转回来。原因我想只需要稍微思考一下就能明白,这里不赘述.
2、应该使用push
进行添加,因为栈的定义是后进先出,add有歧义,无关性能,单纯的觉得用push更好
3、应该使用ArrayDeque
代替Stack
为什么要用ArrayDeque代替Stack?
一句话,stack
性能低,stack
继承自Vector
,vector
对函数加了锁;
关于ArrayDeque
和Stack
的更多知识请参阅 https://blog.csdn.net/deng_hui_long/article/details/120639629。
最后,推荐使用java解答(我只会java和kotlin),根据leetcode的提交记录显示,java的运输速度和内存消耗都远远小于kotlin
最终的代码
class CQueue {
public CQueue() {
}
private Deque<Integer> inQ = new ArrayDeque<Integer>();
private Deque<Integer> outQ = new ArrayDeque<Integer>();
public void appendTail(int value) {
inQ.push(value);
}
public int deleteHead() {
if (outQ.isEmpty()) {
if (inQ.isEmpty()) {
return -1;
}
while (!inQ.isEmpty()) {
outQ.push(inQ.pop());
}
}
return outQ.pop();
}
}