力扣题解： 面试题 03.04. 化栈为队

题目

实现一个MyQueue类，该类用两个栈来实现一个队列。
示例：
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.peek(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
说明：
你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。
假设所有操作都是有效的 （例如，一个空的队列不会调用 pop 或者 peek 操作）。

解题思路

题目要求使用2个栈实现队列

• 队列：先进先出
• 栈：后进先出
  • 1个栈的话，先入栈再出栈的数据是倒序的。例如abc入栈，出栈后是cba
  • 2个栈的话，将倒序的数据再倒序一次，变为正序（实现了先进先出）

代码

import java.util.Deque;
import java.util.LinkedList;

public class MyQueue {
  Deque<Integer> offer, poll;

  public MyQueue() {
    // 用于入队
    offer = new LinkedList<>();
    // 用于出队
    poll = new LinkedList<>();
  }

  public void push(int x) {
    offer.push(x);
  }

  public int pop() {
    prepare();
    return poll.pop();
  }

  public int peek() {
    prepare();
    return poll.peek();
  }

  public boolean empty() {
    return offer.isEmpty() && poll.isEmpty();
  }

  private void prepare() {
    // 如果用于"出队"的栈为空
    if (poll.isEmpty()) {
      // 将用于"入队"的栈中元素pop出来，push到用于"出队"的栈中
      // push进来的元素为倒序，pop出去后变为正序
      while (!offer.isEmpty()) {
        poll.push(offer.pop());
      }
    }
  }
}

题目来源：力扣（LeetCode）