leecode 225. 用队列实现栈-CSDN博客

本文链接：https://blog.csdn.net/maocaicai/article/details/122750557

栈的常用方法在题目中，队列LinkedList的常用方法为poll() , add() , remove();本题和232. 用栈实现队列是栈和队列的相互实现经典题目，考察对栈和队列的理解。（记住队列实现栈只需要在push方法上改变即可，其他不变。）

用队列实现栈，首先想到用两个队列，只需要改变push()方法，其他复用队列方法即可。一个主队列，一个辅助队列，先向辅助队列add新元素，然后判断主队列是否为空，若为空则直接将辅助队列变为主队列，若不为空，则将主队列的元素poll并add辅助队列中，然后将辅助队列变为主队列，这样此队列可直接实现为栈。核心代码如下：

   public void push(int x) {
        q2.add(x);
        while (!q1.isEmpty()) {
            q2.add(q1.poll());
        }
        Queue<Integer> temp = q1;
        q1 = q2;
        q2 = temp;

    }

另一种方法是使用一个队列实现，直接在原队列上改即可：

  public void push(int x) {
        int n = queue.size();
        queue.offer(x);
        for (int i = 0; i < n; i++) {
            queue.offer(queue.poll());
        }
    }

最终实现：

class MyStack {
    Queue<Integer> q1;
    Queue<Integer> q2;
    public MyStack() {
        q1 = new LinkedList<>();
        q2 = new LinkedList<>();
    }
    
    public void push(int x) {
        q2.add(x);
        while (!q1.isEmpty()) {
            q2.add(q1.poll());
        }
        Queue<Integer> temp = q1;
        q1 = q2;
        q2 = temp;

    }
    
    public int pop() {
        return q1.poll();
    }
    
    public int top() {
        return q1.peek();
    }
    
    public boolean empty() {
        return q1.isEmpty() && q2.isEmpty();
    }

}

/**
 * Your MyStack object will be instantiated and called as such:
 * MyStack obj = new MyStack();
 * obj.push(x);
 * int param_2 = obj.pop();
 * int param_3 = obj.top();
 * boolean param_4 = obj.empty();
 */