题目:
请你仅使用两个栈实现先入先出队列。
队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
- void push(int x) 将元素 x 推到队列的末尾
- int pop() 从队列的开头移除并返回元素
- int peek() 返回队列开头的元素
- boolean empty() 如果队列为空,返回 true ;否则,返回 false
说明:
你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
进阶:
你能否实现每个操作均摊时间复杂度为 O(1) 的队列?换句话说,执行 n 个操作的总时间复杂度为 O(n) ,即使其中一个操作可能花费较长时间。
示例:
输入:
- [“MyQueue”, “push”, “push”, “peek”, “pop”, “empty”]
- [[], [1], [2], [], [], []]
输出:
- [null, null, null, 1, 1, false]
解释:
- MyQueue myQueue = new MyQueue();
- myQueue.push(1); // queue is: [1]
- myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
- myQueue.peek(); // return 1
- myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false
思路:
准备两个栈 inStack、outStack
- 如果掉用push,直接就push进去
- 如果调用 pop,需要判断 outStack 是否为空,如果不是空就将 inStack 颠倒给到 outStack,返回 outStack 第一项(pop)
- 调用 peek,需要判断 outStack 是否为空,不是空就将 inStack 颠倒给到 outStack,返回 outStack 最后一项
- 调用 empty,需要判断 inStack 和 outStack 都不是空
- 做一个颠倒函数 ```javascript // new的时候生成两个栈,一个是进的一个是出的 var MyQueue = function () { this.inStack = [] this.outStack = [] };
MyQueue.prototype.push = function (x) { this.inStack.push(x) };
// this.outStack 栈为空, MyQueue.prototype.pop = function () { if (!this.outStack.length) { this.inAndOut() } return this.outStack.pop(); };
MyQueue.prototype.peek = function () { if (!this.outStack.length) { this.inAndOut() } return this.outStack[this.outStack.length - 1] };
MyQueue.prototype.empty = function () { return this.inStack.length === 0 }; // 循环遍历颠倒过来 MyQueue.prototype.inAndOut = function () { while (this.inStack.length) { this.outStack.push(this.inStack.pop()); } }; ```
