【每日算法】用两个栈实现一个队列

115 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第34天,点击查看活动详情

前言

今天是2022的最后一天,看完B站的跨年晚会闲来无事上去 leetcode 练练手,也算是今年的收尾吧

题目

剑指 Offer 09. 用两个栈实现队列 - 力扣(LeetCode)

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )

示例 1:

输入:
["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]
[[],[3],[],[],[]]
输出:[null,null,3,-1,-1]

示例 2:

输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]

分析

老实说,这个题目拿到手里的时候着实有点懵,主要是这两个示例实在是太抽象了

这里先简单解释一下这个示例想表达什么

首先示例给了2个数组,一个数组代表相关的代码操作,一个数组表示相关操作所需要传递的参数,最后就是执行结果

  • ["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]
  • [[],[3],[],[],[]]

上下两个数组一一对应,依次执行

  • CQueue:初始化队列;不需要参数;没有输出结果,也就是 null
  • appendTail: 尾部插入数据;参数是3;插入操作不需要输出结果,也就是 null
  • deleteHead: 头部删除数据; 不需要参数;删除操作输出删除的元素,也就是 3
  • deleteHead:头部删除数据; 不需要参数;删除操作输出删除的元素,但是由于队列里已经没有数据,所以输出 -1

好了,示例的意思讲解完成,接下来我们要来做题了

思路

题目已经给了提示,用两个栈实现一个队列,那么我们的代码的所有功能一定是基于这两个栈实现的

  • 首先,什么是栈

栈是一种数据结构,特点是先进,后出(即只能在末尾进行操作)

如果用javascript的数组去模拟栈的话,那么就是一个只有push(从末尾添加元素)和pop(从末尾删除元素)2个方法的数组

  • 要实现的队列

队列这里题目也做了解释,就是只能在尾部插入appendTail,在头部删除deleteHead

理解完题目的目标之后,我们就来编写代码了

实现

var CQueue = function() {
    this.stack1 = []
    this.stack2 = []
};
​
/** 
 * @param {number} value
 * @return {void}
 */
CQueue.prototype.appendTail = function(value) {
    this.stack1.push(value)
};
​
/**
 * @return {number}
 */
CQueue.prototype.deleteHead = function() {
    if (this.stack2.length) {
        return this.stack2.pop()
    } else {
        while(this.stack1.length){
            this.stack2.push(this.stack1.pop())
        }
        if (this.stack2.length) {
            return this.stack2.pop()
        } else {
            return -1
        }
    }
};
​
/**
 * Your CQueue object will be instantiated and called as such:
 * var obj = new CQueue()
 * obj.appendTail(value)
 * var param_2 = obj.deleteHead()
 */