用队列实现栈

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

    push(x) -- 元素 x 入栈
    pop() -- 移除栈顶元素
    top() -- 获取栈顶元素
    empty() -- 返回栈是否为空

注意:

    你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。
    你所使用的语言也许不支持队列。 你可以使用 list 或者 deque（双端队列）来模拟一个队列 , 只要是标准的队列操作即可。
    你可以假设所有操作都是有效的（例如, 对一个空的栈不会调用 pop 或者 top 操作）。

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

 

解题思路

可以使用两个队列A、B来实现一个栈的效果。

入栈：把新的元素放入 A 队列中入队；

出栈：把 A 队列中的元素转入 B ，当 A 中只剩一个元素的时候，就把最后的元素直接入队列即可，然后交换 A 和 B 的身份，保证入栈的元素始终是进入 A 中 

取栈顶元素：和出栈类似

判定为空：A 和 B 都为空 

 

代码实现

 
class MyStack {
    /** Initialize your data structure here. */
    private Queue<Integer> A = new LinkedList<>();
    private Queue<Integer> B = new LinkedList<>();

    /** Push element x onto stack. */
    public void push(int x) {
        A.offer(x);

    }

    /** Removes the element on top of the stack and returns that element. */
    public Integer pop() {
        if (empty()){
            return null;
        }
        while (A.size() > 1){
            Integer front = A.poll();
            B.offer(front);
        }
        //此时 A 就剩下 1 个元素
        int ret = A.poll();
        swapAB();
        return ret;
    }
    private void swapAB(){
        Queue<Integer> tmp = A;
        A = B;
        B = tmp;
    }

    /** Get the top element. */
    public Integer top() {
        if (empty()){
            return null;
        }
        while (A.size() > 1){
            Integer front = A.poll();
            B.offer(front);
        }
        //此时 A 就剩下 1 个元素
        int ret = A.poll();
        //pop  和 top 为以的区别就是这句话
        B.offer(ret);
        swapAB();
        return ret;
    }

    /** Returns whether the stack is empty. */
    public boolean empty() {
        return A.isEmpty() && B.isEmpty();
    }
}