题目:

请你仅使用两个栈实现先入先出队列。
队列应当支持一般队列支持的所有操作(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()); } }; ```