【Leetcode栈与队列 232. 用栈实现队列】C++ 「重要基础」

🎈个人主页：算法诗人
✨收录专栏：算法诗人Leetcode揭秘之旅
🎉欢迎 👍点赞✍评论⭐收藏
🤝希望我的文章能对你们有所帮助，有不足的地方请在评论区留言指正，大家一起学习交流！🤗

---

---

Leetcode题目链接

232. 用栈实现队列

---

一、题目描述

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：

• void push(int x) 将元素 x 推到队列的末尾
• int pop() 从队列的开头移除并返回元素
• int peek() 返回队列开头的元素
• boolean empty() 如果队列为空，返回 true ；否则，返回 false

说明：

• 你 只能 使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
• 你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。

示例 1：
输入：
[“MyQueue”, “push”, “push”, “peek”, “pop”, “empty”]
[[], [1], [2], [], [], []]
输出：
[null, null, null, 1, 1, false]

解释：
MyQueue myQueue = new MyQueue();
myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1
myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false

提示：
1 <= x <= 9
最多调用 100 次 push、pop、peek 和 empty
假设所有操作都是有效的 （例如，一个空的队列不会调用 pop 或者 peek 操作）

进阶：
你能否实现每个操作均摊时间复杂度为 O(1) 的队列？换句话说，执行 n 个操作的总时间复杂度为 O(n) ，即使其中一个操作可能花费较长时间。

二、解题思路

利用两个栈来实现队列，首先我们要明确队列是先进先出，栈是先进后出，所以用一个栈肯定是无法实现队列的。

所以我们就能想到用一个栈stack_push来表示实现的输入，另一个栈stack_pop来实现队列的输出。

这是我们解这道题的思路：

• 入队操作 (push)： 将元素压入 stack_in 栈中，实现了队列的入队操作。
• 出队操作 (pop)： 如果 stack_out 栈为空，就将 stack_in 栈的元素逐个弹出并压入 stack_out 栈，然后从 stack_out 栈顶取出元素，实现了队列的出队操作。
• 查看队头元素 (peek)： 在 peek 操作中，调用 pop 方法获取队头元素，并将这个元素重新压入 stack_out 栈，保持队列状态不变。
• 检查队列是否为空 (empty)： 判断 stack_in 和 stack_out 是否都为空，来确定整个队列是否为空。

三、示例代码

class MyQueue {
public:
    stack<int> stack_in;
    stack<int> stack_out;

    MyQueue() {
        
    }
    
    void push(int x) {
        stack_in.push(x);
    }
    
    int pop() {
        if(stack_out.empty())
        {
            while(!stack_in.empty())
            {
                stack_out.push(stack_in.top());
                stack_in.pop();
            }
        }
        int result = stack_out.top();
        stack_out.pop();
        return result;
    }
    
    int peek() {
        int result = this->pop();
        stack_out.push(result);
        return result;
    }
    
    bool empty() {
        return stack_in.empty() && stack_out.empty();
    }
};

/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue* obj = new MyQueue();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->peek();
 * bool param_4 = obj->empty();
 */

时间复杂度：
入队操作是 O(1) 的，因为直接将元素压入 stack_in。
出队操作在最坏情况下需要将 stack_in 中的所有元素移到 stack_out，因此是 O(N)，其中 N 是队列中的元素个数。
查看队头元素和检查队列是否为空都是 O(1) 的。

空间复杂度：
使用了两个栈，因此空间复杂度是 O(N)，其中 N 是队列中的元素个数。

新手可能不理解的代码（其实是我自己学习到的新知识）

1. queue的pop()和stack的pop()的不同之处

queue的pop()是有返回值的，stack的pop()只是移除了最后进入的元素，而在stack中想找到最后进入的元素的值需要用到top()。

这也是从stack_in中取值存到stack_out分为两个步骤的原因：

	stack_out.push(stack_in.top());
    stack_in.pop();

2. int result = this->pop()怎么理解？

其实这里就是调用了我们自己设计的queue结构中的pop()函数，因为队列的peek()和pop()返回的值是一样的，我们在peek()中就可以复用一下pop()，然后再把pop出去的值再push回去就OK了。

结尾碎碎念
“这次的算法解析只是冰山一角，未来我将持续分享更多有趣的算法问题，并提供更深入的解析。如果你对这个系列感兴趣，不要错过即将到来的更新！点赞、关注、收藏，如果有不同的解法欢迎留下你的评论，你们的支持就是我更新的最大动力。让我们一起探索更多算法的奇妙世界，这不仅是为了迎接面试挑战，更是为了成为在编程领域中游刃有余的专家。感谢大家的支持，期待未来更多的相遇和共同成长！”