LeeCode225:用队列实现栈

题目来源

https://leetcode-cn.com/problems/implement-stack-using-queues/

题目描述

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

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

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

解题思路

• 使用两个队列来实现栈。queue1作为优先输入队列。queue2作为优先输出队列。

实现代码

public class UseQueuesImplmentStack {
    private Queue<Integer> queue1;    //q1作为优先输入队列
    private Queue<Integer> queue2;    //q2作为优先输出队列
    public int size;

    public UseQueuesImplmentStack() {
        queue1 = new LinkedList<Integer>();    //初始化队列
        queue2 = new LinkedList<Integer>();
        size = 0;
    }

    /** Push element x onto stack. */
    public void push(int x) {
        if (queue1.peek() != null) {    //q1队列不为空
            queue1.offer(x);
        } else if (queue2.peek() != null) {    //q2队列不为空
            queue2.offer(x);
        } else {
            queue1.offer(x);    //如果q1,q2都为空，选择q1放入数据
        }
        this.size++;
    }

    /** Removes the element on top of the stack and returns that element. */
    public int pop() {
        int returnVal = 0;
        if (queue2.peek() != null) { //q2不为空，先把q2的size-1个数据给q1。只剩一个数据
            for (int i = 0; i < size - 1; i++) {
                queue1.offer(queue2.poll());
            }
            returnVal = queue2.peek();
            queue2.poll(); // 然后把q2剩得数据出出去
        } else {
            for (int i = 0; i < size - 1; i++) {
                queue2.offer(queue1.poll());
            }
            returnVal = queue1.peek();
            queue1.poll();
        }
        size--;
        return returnVal;
    }

    /** Get the top element. */
    public int top() {
        if (empty()){
            throw new RuntimeException("无数据");
        }
        int temp = 0;
        if (queue2.peek() != null) {    //q2不为空，top一定在q2,
            for (int i = 0; i < size; i++) {
                //将q2的数据搬移前先保存在temp，temp被一直覆盖，最后一个q2数据就是top
                temp = queue2.poll();
                queue1.offer(temp);
            }
        } else{    //q2为空,top在q1
            for (int i = 0; i < size; i++) {
                temp = queue1.poll();
                queue2.offer(temp);
            }
        }
        return temp;
    }

    /** Returns whether the stack is empty. */
    public boolean empty() {
        return this.size == 0;
    }
}