代码随想录算法训练营Day7 | 233.用栈实现队列、225.用队列实现栈

LeeCode 233 用栈实现队列

本题思路：使用两个栈来实现队列，应该怎么做呢？我们通过画图来分析下

• 入队列的时候，直接在 stackin 入
• 出队列的时候，肯定是先出 1 ，此时把，stackin 中全部 弹出到 入到 stackout 中，如果要出队列，直接弹出 stackout 顶部元素就行。（此处，因为 stackout 为空，所以需要将 stackin 所有元素弹入到 stackout 中）
• 此时如果继续入队列 3 就继续往 stackin 中添加
• 此时再要出队列的时候，就是出的2，此时 stackout 中不为空，直接返回 stackout.pop 即可
• 此时再进入队列 4 ，直接进入 stackin
• 后续的逻辑就和前面一样

public 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() {
        // 首先判断 stackout 是否为空
        if (stackout.isEmpty()) {
            // 把 stackin 元素全部放入 stackout中
            while (!stackin.isEmpty()) {
                stackout.push(stackin.pop());
            }
        } 
        return stackout.pop();
    }

    // 查看队首元素
    public int peek() {
        // 首先判断 stackout 是否为空
        if (stackout.isEmpty()) {
            // 把 stackin 元素全部放入 stackout中
            while (!stackin.isEmpty()) {
                stackout.push(stackin.pop());
            }
        } 
        return stackout.peek();
    }

    // 判空操作
    public boolean empty() {
        return (stackin.isEmpty() && stackout.isEmpty()) == true ? true : false;
    }
}

LeetCode 225 用队列实现栈

本题思路：使用 一个队列来模拟实现栈更简单。下面画个图来分析一下

• 进栈操作
• 出栈操作，此时应该出的是 3 ，我们可以将 3 之前的元素 从队头弹出，然后重新加入到队列中，此时队头元素就是 3，直接弹出返回这个元素即可

public class MyStack {

    Queue<Integer> queue;
    int size = 0;

    public MyStack() {
        queue = new ArrayDeque<>();
    }

    // 添加操作，直接入栈即可（入队列）
    public void push(int x) {
        queue.add(x);
        size++;
    }

    // 出栈操作
    public int pop() {
        // 先弹出 size-1 个元素，重新加到队列末尾。
        for (int i = 0; i < size - 1; i++) {
            queue.add(queue.poll());
        }
        size--;
        return queue.poll();
    }

    // 查看栈顶元素
    public int top() {
        // 先弹出 size-1 个元素，重新加入到队列末尾。
        for (int i = 0; i < size - 1; i++) {
            queue.add(queue.poll());
        }
        // 保存下来栈顶元素
        int res = queue.poll();
        queue.add(res);
        return res;
    }

    public boolean empty() {
        return size == 0 ? true : false;
    }
}