1. 题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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 次调用
2. 解题思路
我们知道,栈是一种先进后出的数据结构,队列是一种先进先出的数据结构。想要使用栈来模拟队列的操作,我们就需要两个栈来实现,这两个栈的分工不同,一个负责入栈操作,一个负责出栈的操作,具体的实现逻辑如下:
- 初始化两个栈stackA和stackB,分别负责入栈和出栈操作;
- 如果是入栈操作,就将数据push到stackA栈中;
- 如果是出栈操作,先检查出队列栈stackB中是否有元素,如果有就直接出栈;如果没有就将stackA中元素按照入栈顺序,依次放在stackB栈中,再从B中出栈。
- 如果stackA和stackB的长度都为0,那么直接返回-1。
复杂度分析:
- 时间复杂度:O(1),删除和插入操作的时间复杂度都是O(1);
- 空间复杂度:O(n),需要使用两个栈来存储已有的元素。
3. 代码实现
```typescript var CQueue = function() { this.stackA = [] this.stackB = []
};
/**
- @param {number} value
- @return {void} */ CQueue.prototype.appendTail = function(value) { this.stackA.push(value) };
/**
- @return {number}
*/
CQueue.prototype.deleteHead = function() {
if(this.stackB.length){
}else{return this.stackB.shift()
} };while(this.stackA.length){
this.stackB.push(this.stackA.shift())
}
if(!this.stackB.length){
return -1
}else{
return this.stackB.shift()
}
/**