用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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 次调用
思路:
通过题目可以了解,两个栈,stackOne和stackTwo,是要通过两个 “先进后出” 的栈,实现一个 “先进先出” 的队列。
所以我们考虑队列插入和删除栈顶的元素的方法:
- 我们把让stackOne存放数据。进入队列的数据,都放到stackOne里面
- 在删除栈顶元素的时候,需要删除先进入stackOne的元素
- 这就考虑使用stackTwo,我们把stackOne里的数据都弹出,然后再压入stackTwo中
- 数据就和stackOne中的相反了。
- 所以当stackTwo不为空的时候,就直接出栈,
- 当stackTwo为空的时候,就把stackTwo的数据弹出并压入stackTwo中。
- 这样就可以了
class CQueue {
private stackOne: number[];
private stackTwo: number[];
constructor() {
this.stackOne = [];
this.stackTwo = [];
}
public appendTail(value: number): void {
this.stackOne.push(value);
}
public deleteHead(): number {
if (!this.stackTwo.length) {
if (!this.stackOne.length) {
return -1;
}
while (this.stackOne.length) {
this.stackTwo.push(this.stackOne.pop()!);
}
return this.stackTwo.pop()!;
}
return this.stackTwo.pop()!;
}
}