JAVA——用栈实现队列

问题：请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：
void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空，返回 true ；否则，返回 false

方法：
1.创建两个栈
2.push(int x) 将元素 x 推到队列的末尾：将元素全都放到栈1中。

3.pop() 从队列的开头移除并返回元素：分为两种情况。第一种要考虑两个栈都为空的情况。第二种要考虑在栈2为空的情况下，若栈1不为空则把栈1中要删除的元素，push到栈2中，再pop删除并返回栈顶元素（队头元素）（如图第一行）；栈2不为空的话就直接返回并删除栈顶元素（如图第二行，要先出了已经放入栈2中的元素，才能放入新的元素）。

4.peek() 返回队列开头的元素：不用删除，直接修改3中的pop为peek。
代码：

class MyQueue {
    private Stack<Integer> s1;
    private Stack<Integer> s2;
    
    public MyQueue() {
     s1=new Stack<>();
     s2=new Stack<>();
    }

    public void push(int x) {
        s1.push(x);//指定push到第一个栈中
    }

    //删除队头元素
    public int pop() {
     if(empty()){
         return -1;
     }
     if(s2.empty()){
         while(!s1.empty()){
             s2.push(s1.pop());
         }
     }
     return s2.pop();
    }

    //获取队头元素
    public int peek() {
        if(empty()){
            return -1;
        }
        if(s2.empty()){
            while(!s1.empty()) {
                s2.push(s1.pop());
            }
        }
        return s2.peek();
    }

    public boolean empty() {
     return s1.isEmpty()&&s2.isEmpty();
    }
}