Java 求解用栈实现队列

Posted 2021-12-09 南淮北安

文章目录

• • 一、题目
  • 二、代码

一、题目

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

实现 MyQueue 类：

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

二、代码

class MyQueue {
    Deque<Integer> deque1;
    Deque<Integer> deque2;

    /**
     * Initialize your data structure here.
     */
    public MyQueue() {
        deque1 = new LinkedList<>();
        deque2 = new LinkedList<>();
    }

    /**
     * Push element x to the back of queue.
     */
    public void push(int x) {
        deque1.push(x);
    }

    /**
     * Removes the element from in front of queue and returns that element.
     */
    public int pop() {
        if (deque2.isEmpty()) {
            while (!deque1.isEmpty()) {
                deque2.push(deque1.pop());
            }
        }
        return deque2.pop();
    }

    /**
     * Get the front element.
     */
    public int peek() {
        if (deque2.isEmpty()) {
            while (!deque1.isEmpty()) {
                deque2.push(deque1.pop());
            }
        }
        return deque2.peek();
    }

    /**
     * Returns whether the queue is empty.
     */
    public boolean empty() {
        return deque2.isEmpty() && deque1.isEmpty();
    }
}