代码随想录算法训练营Day9|LC232 用栈实现队列&LC225 用队列实现栈

最新推荐文章于 2024-09-15 19:46:31 发布

include_stdio.h

最新推荐文章于 2024-09-15 19:46:31 发布

阅读量314

点赞数 4

文章标签： java 开发语言

本文链接：https://blog.csdn.net/qq_62428331/article/details/136383552

版权

本文介绍了如何使用Java中的栈和队列实现对方的功能，包括用栈模拟队列的push、pop和peek操作，以及用队列实现栈的push、pop和top操作，展示了数据结构在问题转换中的应用。

摘要由CSDN通过智能技术生成

一句话总结：栈和队列的基础用法，思路不难。

原题链接：232 用栈实现队列

原理就是利用两个栈实现一个队列，难点是push()操作。简单来说，在输出栈为空时将输入栈的元素依次输出到输出栈，这样输出栈从栈顶到栈底就是队列的从队头到队尾的顺序。

class MyQueue {
    Deque<Integer> inStack;
    Deque<Integer> outStack; 

    public MyQueue() {
        inStack = new LinkedList<>();
        outStack = new LinkedList<>();
    }
    
    public void push(int x) {
        inStack.push(x);
    }
    
    public int pop() {
        if (outStack.isEmpty()) inToOut();
        return outStack.pop();
    }
    
    public int peek() {
        if (outStack.isEmpty()) inToOut();
        return outStack.peek();
    }
    
    public boolean empty() {
        return inStack.isEmpty() && outStack.isEmpty();
    }

    void inToOut() {
        while (!inStack.isEmpty()) outStack.push(inStack.pop());
    }
}

其中需要另写一个函数在输入栈非空时将输入栈元素压入输出栈。

原题链接：225 用队列实现栈

一个队列即可解决。方法是加入新元素时先记录当前队列内元素数，然后先将新元素加入队尾，再将加入新元素前的元素依次出队再依次入队，即可完成栈的操作。

class MyStack {
    Deque<Integer> queue;

    public MyStack() {
        queue = new LinkedList<>();
    }
    
    public void push(int x) {
        int n = queue.size();
        queue.offer(x);
        for (int i = 0; i < n; ++i) {
            queue.offer(queue.poll());
        }
    }
    
    public int pop() {
        return queue.poll();
    }
    
    public int top() {
        return queue.peek();
    }
    
    public boolean empty() {
        return queue.isEmpty();
    }
}