【力扣】232. 用栈实现队列

一、题目描述：

使用栈实现队列的下列操作：

• push(x) – 将一个元素放入队列的尾部。
• pop() – 从队列首部移除元素。
• peek() – 返回队列首部的元素。
• empty() – 返回队列是否为空。

示例:

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 操作）。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/implement-queue-using-stacks
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路：

创建两个栈A和B。栈A专门用来入队列，栈B专门用来出队列。
1、入队列：把B中的所有元素转到A中，让新元素入栈A即可。
2、出队列：把A中的所有元素转到B中，让B的栈顶元素出栈即可。
3、取队首元素：把A中的所有元素转到B中，B的栈顶元素就是队首元素，让B的栈顶元素出栈即可。
4、判空：A和B都为空。

三、代码描述：

class MyQueue {
    Stack<Integer> A=new Stack<>();
    Stack<Integer> B=new Stack<>();
    
    public void push(int x) {
        if(!B.isEmpty()){
            Integer temp=B.pop();
            A.push(temp);
        }
        A.push(x);
    }

    public Integer pop() {
      if(empty()){
          return null;
      }
      while(!A.isEmpty()){
          Integer temp=A.pop();
          B.push(temp);
      }
      return B.pop();
    }

    public Integer peek() {
        if(empty()){
            return null;
        }
        while(!A.isEmpty()){
            Integer temp=A.pop();
            B.push(temp);
        }
        return B.peek();
    }

    public boolean empty() {
      return A.isEmpty()&&B.isEmpty();
    }
}