基于java用队列实现栈

问题描述

请你仅使用两个队列实现一个后入先出（LIFO）的栈，并支持普通栈的全部四种操作（push、top、pop 和 empty）。

实现 MyStack 类：

void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的，返回 true ；否则，返回 false.

注意：

你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size和 is empty
这些操作。 你所使用的语言也许不支持队列。 你可以使用 list （列表）或者 deque（双端队列）来模拟一个队列 ,只要是标准的队列操作即可。

实例

原题OJ链接

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

解答

class MyStack {

    Queue<Integer> que1;
    Queue<Integer> que2;

    public MyStack() {
        que1 = new LinkedList<>();
        que2 = new LinkedList<>();
    }

    public void push(int x) {
        if(!que1.isEmpty()){
            que1.offer(x);
        }else{
            if (!que2.isEmpty()){
                que2.offer(x);
            }
            else{
                que1.offer(x);
            }
        }
    }

    public int pop() {
        if(!que1.isEmpty()){
            /*for (int i = 0; i < que1.size()-1; i++) {
                int ret = que1.poll();
                que2.offer(ret);
            }*/
            //这样写是错误的，因为que1.size()随着弹出元素是会变化的
            int size = que1.size();
            for (int i = 0; i < size-1; i++) {
                int ret = que1.poll();
                que2.offer(ret);

            }
            return que1.poll();
        }
        else {
            if(!que2.isEmpty()){
                int size = que2.size();
                for (int i = 0; i < size-1; i++) {
                    int ret = que2.poll();
                    que1.offer(ret);
                }
                return que2.poll();
            }else{
                return -1;
            }
        }
    }

    public int top() {

        if(!que1.isEmpty()){
            int size = que1.size();
            int ret = 0;
            for (int i = 0; i < size; i++) {
                ret = que1.poll();
                que2.offer(ret);
            }
            return ret;
        }
        else {
            if(!que2.isEmpty()){
                int size = que2.size();
                int ret = 0;
                for (int i = 0; i < size; i++) {
                    ret = que2.poll();
                    que1.offer(ret);
                }
                return ret;
            }else{
                return -1;
            }
        }

    }

    public boolean empty() {
        return que1.isEmpty() && que2.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();
 */

测试结果