225. 用队列实现栈

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

题目要求

• 使用栈实现队列的下列操作：
  • push(x) – 将一个元素放入队列的尾部。
  • pop() – 从队列首部移除元素。
  • peek() – 返回队列首部的元素。
  • empty() – 返回队列是否为空。

两个栈实现出入队

    class MyQueue {
        Stack<Integer> A;
        Stack<Integer> B;

        public MyQueue() {
            A = new Stack<>();
            B = new Stack<>();
        }

        public void push(int x) {
            while (!A.isEmpty()) {
                B.push(A.pop());
            }
            B.push(x);
            while (!B.isEmpty()) {
                A.push(B.pop());
            }
        }

        public int pop() {
            return A.pop();
        }

        public int peek() {
            return A.peek();
        }

        public boolean empty() {
            return A.isEmpty();
        }
    }

维护进栈和出栈

    class MyQueue {

        Stack<Integer> stackIn;
        Stack<Integer> stackOut;

        public MyQueue() {
            stackIn = new Stack<>(); // 负责进栈
            stackOut = new Stack<>(); // 负责出栈
        }

        public void push(int x) {
            stackIn.push(x);
        }

        public int pop() {
            dumpStackIn();
            return stackOut.pop();
        }

        public int peek() {
            dumpStackIn();
            return stackOut.peek();
        }

        public boolean empty() {
            return stackIn.isEmpty() && stackOut.isEmpty();
        }

        // 如果stackOut为空，那么将stackIn中的元素全部放到stackOut中
        private void dumpStackIn() {
            if (!stackOut.isEmpty()) return;
            while (!stackIn.isEmpty()) {
                stackOut.push(stackIn.pop());
            }
        }
    }