leetcode-用两个栈实现队列

2022-10-11  本文已影响0人  王灵

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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继承自Vectorvector对函数加了锁;
关于ArrayDequeStack的更多知识请参阅 https://blog.csdn.net/deng_hui_long/article/details/120639629

最后,推荐使用java解答(我只会java和kotlin),根据leetcode的提交记录显示,java的运输速度和内存消耗都远远小于kotlin


20221011164239.jpg

最终的代码

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();
    }
}
上一篇下一篇

猜你喜欢

热点阅读