Leetcode--Java--225. 用队列实现栈

题目描述

请你仅使用两个队列实现一个后入先出（LIFO）的栈，并支持普通栈的全部四种操作（push、top、pop 和 empty）。
实现 MyStack 类：
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的，返回 true ；否则，返回 false 。

样例描述

输入：
["MyStack", "push", "push", "top", "pop", "empty"]
[[], [1], [2], [], [], []]
输出：
[null, null, null, 2, 2, false]

解释：
MyStack myStack = new MyStack();
myStack.push(1);
myStack.push(2);
myStack.top(); // 返回 2
myStack.pop(); // 返回 2
myStack.empty(); // 返回 False

思路

双队列

1. 一个队列作为栈，另外一个队列作为辅助队列，用于调换元素的顺序。（因为栈里面是后进先出）。
2. 入栈操作时，先将元素都加入到辅助队列，再将主队列的元素加入到辅助队列，然后交换两个队列的元素，就能够保证在主队列里的元素都是符合后到先出的。
3. 容器也能用交换数的方法交换。

代码

class MyStack {
    
    
     Queue<Integer> queue1, queue2;
    public MyStack() {
    
    
       queue1 = new LinkedList<>();
       queue2 = new LinkedList<>();
    }
    
    public void push(int x) {
    
    
        //先将元素加入到辅助队列
        queue2.offer(x);
        //将主队列的元素全部加入到辅助队里的末尾
        while (!queue1.isEmpty()) {
    
    
            queue2.offer(queue1.poll());
        }
        //交换两个队列，准备下一轮新元素插入时的迭代
        Queue<Integer> t = queue1;
        queue1 = queue2;
        queue2 = t;
    }
    
    public int pop() {
    
    
       return queue1.poll();
    }
    
    public int top() {
    
    
       return queue1.peek();
    }
    
    public boolean empty() {
    
    
       return queue1.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();
 */